Peter Hawkins 362fb7ae9d Remove code to support jaxlib < 0.5.3. ...

The new xla_extension_version is 320.

PiperOrigin-RevId: 738522486

2025-03-19 13:40:04 -07:00

..

__init__.py

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

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

_export.py

Remove code to support jaxlib < 0.5.3.

2025-03-19 13:40:04 -07:00

serialization_generated.py

support e2m1fn

2025-03-05 17:44:34 +00:00

serialization.fbs

support e2m1fn

2025-03-05 17:44:34 +00:00

serialization.py

support e2m1fn

2025-03-05 17:44:34 +00:00

shape_poly_decision.py

[shape_poly] Improve threefry with symbolic shapes

2025-01-07 09:10:04 +02:00

shape_poly.py

refactor: move jnp.einsum impl into its own submodule

2025-02-12 09:05:30 -08:00