Parallelization via sharding (multi-device) composed with vmap (across memory in single device)#1773
Open
unalmis wants to merge 28 commits into
Open
Parallelization via sharding (multi-device) composed with vmap (across memory in single device)#1773unalmis wants to merge 28 commits into
unalmis wants to merge 28 commits into