Parallelization via sharding (multi-device) composed with vmap (across memory in single device) - #1773
Open
unalmis wants to merge 37 commits into
Open
Parallelization via sharding (multi-device) composed with vmap (across memory in single device)#1773unalmis wants to merge 37 commits into
unalmis wants to merge 37 commits into
Commits
Commits on Dec 16, 2025
Commits on Feb 27, 2026
Commits on Mar 29, 2026
Commits on Apr 1, 2026
Commits on Apr 8, 2026
Commits on Apr 11, 2026
Commits on May 15, 2026
- committed
- committed
- committed
- committed
Commits on May 18, 2026
Commits on May 19, 2026
Commits on Jun 10, 2026
Commits on Jun 19, 2026
Commits on Jun 20, 2026
Commits on Jul 20, 2026
- committed
- committed
- committed
- authored
- authored
- authored
- committed
- authored