Release new version. Add construct_maximal_shardings to public interface.

PiperOrigin-RevId: 796546347

Author: cpgaffney1

checkpoint/CHANGELOG.md

@@ -7,6 +7,8 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 
 ## [Unreleased]
 
+## [0.11.23] - 2025-08-18
+
 ### Added
 
 - Add support for sharded loading from Safetensors checkpoints onto a JAX mesh.

checkpoint/orbax/checkpoint/_src/arrays/sharding.py

@@ -94,7 +94,16 @@ def _construct_maximal_sharding(
 
 
 def construct_maximal_shardings(abstract_state: PyTree) -> PyTree:
-  """Construct a sharding that partitions each array as much as possible."""
+  """Construct a sharding that partitions each array as much as possible.
+
+  This method is subject to change and should not be considered stable.
+
+  Args:
+    abstract_state: PyTree of jax.ShapeDtypeStruct.
+
+  Returns:
+    PyTree of jax.sharding.Sharding.
+  """
   shardings = jax.tree.map(_construct_maximal_sharding, abstract_state)
 
   total_size = 0

checkpoint/orbax/checkpoint/arrays.py

@@ -22,3 +22,7 @@
     Index,
     Shape,
 )
+
+from orbax.checkpoint._src.arrays.sharding import (
+    construct_maximal_shardings,
+)

checkpoint/orbax/checkpoint/version.py

@@ -17,7 +17,7 @@
 # A new PyPI release will be pushed everytime `__version__` is increased.
 # Also modify version and date in CHANGELOG.
 # LINT.IfChange
-__version__ = '0.11.22'
+__version__ = '0.11.23'
 # LINT.ThenChange(//depot//orbax/checkpoint/CHANGELOG.md)