rocm_jax/array_serialization at 1bd5fd2a52b6ee191b5f80b67d4e10d36f3ef912 - rocm_jax - Gitea For EOELAB

mirrors/rocm_jax

mirror of https://github.com/ROCm/jax.git synced 2025-04-27 06:46:07 +00:00

History

Yash Katariya 1bd5fd2a52 Add serialize_with_paths and deserialize_with_paths API to GlobalAsyncCheckpointManager

PiperOrigin-RevId: 555050522

2023-08-08 22:47:27 -07:00

..

__init__.py

Rename gda_serialization to array_serialization but keep gda_serialization around until it is included in a jax release so that OSS projects can be moved to array_serialization

2023-03-31 18:07:51 -07:00

serialization_test.py

Add serialize_with_paths and deserialize_with_paths API to GlobalAsyncCheckpointManager

2023-08-08 22:47:27 -07:00

serialization.py

Add serialize_with_paths and deserialize_with_paths API to GlobalAsyncCheckpointManager

2023-08-08 22:47:27 -07:00