mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-24 07:46:06 +00:00
Code Issues Packages Projects Releases Wiki Activity
rocm_jax/jax/_src/export
History
Sergei Lebedev 1ac6b762dd Ensured that JAX type checks under pytype on Python 3.12 
Some errors uncovered by pytype look genuine and need to be revisited in
the in the future.

PiperOrigin-RevId: 704268742
2024-12-09 06:53:08 -08:00
..
__init__.py
[export] Move the export implementation to jax._src.export.
2024-06-09 08:59:50 -07:00
_export.py
Activate Tridiagonal Reduction to XLA's FFI
2024-12-09 04:36:59 -08:00
serialization_generated.py
Add float8_e4m3 and float8_e3m4 types support
2024-11-08 18:58:31 +00:00
serialization.fbs
Add float8_e4m3 and float8_e3m4 types support
2024-11-08 18:58:31 +00:00
serialization.py
Add float8_e4m3 and float8_e3m4 types support
2024-11-08 18:58:31 +00:00
shape_poly_decision.py
[shape_polyO] Improve handling of equality shape constraints
2024-09-06 13:55:38 +03:00
shape_poly.py
Ensured that JAX type checks under pytype on Python 3.12
2024-12-09 06:53:08 -08:00