Work toward parallelization via sharding (multi-device) composed with vmap (across memory in single device) - #1869
Merged
Commits
Commits on Jun 11, 2025
- committed
Commits on Jun 12, 2025
Commits on Jun 13, 2025
Commits on Jun 14, 2025
Commits on Jun 15, 2025
- committed
Commits on Jun 29, 2025
Commits on Jul 1, 2025
Commits on Jul 24, 2025
- committed
- committed
Commits on Jul 25, 2025
- committed
Commits on Jul 27, 2025
- committed
Commits on Jul 30, 2025
Commits on Aug 3, 2025
Commits on Aug 4, 2025
Commits on Aug 13, 2025
- committed
- committed
- committed
