orbax
orbax copied to clipboard
google
Add support for merging array groups across multiple devices, with uneven sharding between devices.
Add support for merging array groups across multiple devices, with uneven sharding between devices.
![copybara-service[bot] avatar](https://avatars.githubusercontent.com/in/44061?v=4 "Published by a pub.dev verified publisher"){kind=small}