mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-24 10:26:06 +00:00
Code Issues Packages Projects Releases Wiki Activity
rocm_jax/jax/_src/export
History
Paweł Paruzel d474feda9e Activate Tridiagonal Reduction to XLA's FFI 
Additionally, created a missing backward compatibility test for the old LAPACK kernels of Tridiagonal Reduction.

PiperOrigin-RevId: 704234350
2024-12-09 04:36:59 -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
Merge pull request #25252 from gnecula:poly_power
2024-12-06 05:54:18 -08:00