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

# Save Sharded State

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