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 05:46:06 +00:00
Code
Issues
Packages
Projects
Releases
Wiki
Activity
rocm_jax
/
jax
/
_src
/
export
History
jax authors
ea63aeab01
Merge pull request
#25442
from jakevdp:raise-to-shaped
...
PiperOrigin-RevId: 705556199
2024-12-12 10:43:17 -08:00
..
__init__.py
[export] Move the export implementation to jax._src.export.
2024-06-09 08:59:50 -07:00
_export.py
Cleanup: remove uses of no-op raise_to_shaped
2024-12-12 09:49:06 -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_poly] Improve reasoning for >= in presence of == constraints.
2024-12-11 10:51:49 +01:00
shape_poly.py
Merge pull request
#25442
from jakevdp:raise-to-shaped
2024-12-12 10:43:17 -08:00