Source <gh-file:examples/offline_inference/load_sharded_state.py>.

# Load Sharded State

:::{literalinclude} ../../../../examples/offline_inference/load_sharded_state.py
:language: py
:::