Jake VanderPlas 6e837da326 Document jax.export serialization version numbers

2024-06-12 12:44:42 -07:00

..

__init__.py

[export] Move the export implementation to jax._src.export.

2024-06-09 08:59:50 -07:00

_export.py

Document jax.export serialization version numbers

2024-06-12 12:44:42 -07:00

serialization_generated.py

[export] Move the export implementation to jax._src.export.

2024-06-09 08:59:50 -07:00

serialization.fbs

[export] Cleaned up types of [in|out]_shardings

2024-06-11 13:46:44 +02:00

serialization.py

[export] Cleaned up types of [in|out]_shardings

2024-06-11 13:46:44 +02:00

shape_poly_decision.py

[shape_poly] Move to jax._src in preparation for adding to AOT APIs.

2024-06-04 22:03:24 -07:00

shape_poly.py

[export] Add documentation for jax.export

2024-06-12 19:44:47 +02:00