Sergei Lebedev 0ff234049b Removed trivial docstrings from JAX tests ...

These docstrings do not make the tests any more clear and typically just duplicate the test module name.

PiperOrigin-RevId: 737611977

2025-03-17 07:49:37 -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

BUILD

Remove tensorstore dependency from //jax/experimental/array_serialization:serialization in OSS (see https://github.com/google/tensorstore/issues/218)

2025-02-26 14:47:16 -08:00

serialization_test.py

Removed trivial docstrings from JAX tests

2025-03-17 07:49:37 -07:00

serialization.py

[JAX] Fix a small bug if shardings is tuple.

2024-12-18 09:30:32 -08:00