Add support for merging array groups across multiple devices, with un…

…even sharding between devices. PiperOrigin-RevId: 716369704
google · Jan 17, 2025 · 71b0f23 · 71b0f23
1 parent 6ed02f3
commit 71b0f23
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -9,6 +9,8 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 
 ### Added
 - Local type handler registries.
+- Added support for merging array groups across multiple devices, with uneven
+sharding between devices.
 
 ### Changed
 - The PyPi `orbax` package is deprecated in favor of domain-specific namespace