This website requires JavaScript.
Explore
Help
Sign In
mirrors
/
rocm_jax
Watch
1
Star
0
Fork
0
You've already forked rocm_jax
mirror of
https://github.com/ROCm/jax.git
synced
2025-04-24 10:16:07 +00:00
Code
Issues
Packages
Projects
Releases
Wiki
Activity
rocm_jax
/
jax
/
_src
/
export
History
Sergei Lebedev
78da9fa432
Add float8_e4m3 and float8_e3m4 types support
2024-11-08 18:58:31 +00:00
..
__init__.py
[export] Move the export implementation to jax._src.export.
2024-06-09 08:59:50 -07:00
_export.py
[export] Cleanup in the export module.
2024-11-01 22:56:44 -07: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
[shape_polyO] Improve handling of equality shape constraints
2024-09-06 13:55:38 +03:00