rocm_jax/array_serialization at ef65ba8f32ee185ee8a473575fcad07e4cd28acc - rocm_jax - Gitea For EOELAB

mirrors/rocm_jax

mirror of https://github.com/ROCm/jax.git synced 2025-04-26 04:46:08 +00:00

History

Yash Katariya 0d57330fe0 Add mitigation techniques in the error message when a barrier timeout occurs.

PiperOrigin-RevId: 577214081

2023-10-27 08:57:39 -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 mitigation techniques in the error message when a barrier timeout occurs.

2023-10-27 08:57:39 -07:00