mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-24 16:26:06 +00:00
Code Issues Packages Projects Releases Wiki Activity
rocm_jax/jax/_src/export
History
Jake VanderPlas 955e7c4793 Internal: avoid adding _DimExpr to dtypes._weak_types 
This causes problems because internal code assumes it will not be modified. We replace this with an internal registration mechanism.

PiperOrigin-RevId: 721000907
2025-01-29 09:11:02 -08: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
Internal: avoid adding _DimExpr to dtypes._weak_types
2025-01-29 09:11:02 -08:00