wenscarl 638c6ae046 Add e8m0fnu support by conditional dtype.

2025-01-22 21:57:43 +00:00

..

__init__.py

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

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

_export.py

Activate FFI implementation of tridiagonal reduction on GPU.

2025-01-10 09:28:15 -08:00

serialization_generated.py

Add e8m0fnu support by conditional dtype.

2025-01-22 21:57:43 +00:00

serialization.fbs

Add e8m0fnu support by conditional dtype.

2025-01-22 21:57:43 +00:00

serialization.py

Add e8m0fnu support by conditional dtype.

2025-01-22 21:57:43 +00:00

shape_poly_decision.py

[shape_poly] Improve threefry with symbolic shapes

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

shape_poly.py

[shape_poly] Remove old non_negative support.

2025-01-06 07:36:11 -08:00