Easy tutorial for GDA Serialization?

[sh0416]

I wandered around the web and found that tensorstore is the best way to use load and save for sharded state.

#15257

https://ai.googleblog.com/2022/09/tensorstore-for-high-performance.html

This blog post provided a use case using tensorstore and jax to load and save checkpoint.

https://github.com/google/jax/blob/640e15fe070a887197143b76a19a3dced816c8df/jax/experimental/gda_serialization/serialization.py

https://github.com/google-research/t5x/blob/d34fff62ddc71c86b680300f8dca198f9db7b246/t5x/checkpoints.py

However, due to the limited knowledge of myself, I couldn't understand their code easily.

Is there any other material for understanding the concept and containing short code snippet to apply for me?

Thank you