shuw c099e8081d support e2m1fn

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

..

__init__.py

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

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

_export.py

#sdy support JAX export tests when Shardy is enabled.

2025-03-03 04:57:06 -08: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