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-22 22:06:05 +00:00
Code
Issues
Packages
Projects
Releases
Wiki
Activity
rocm_jax
/
jax
/
_src
/
export
History
Dan Foreman-Mackey
39ce7916f1
Activate FFI implementation of tridiagonal reduction on GPU.
...
PiperOrigin-RevId: 714078036
2025-01-10 09:28:15 -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 float8_e4m3 and float8_e3m4 types support
2024-11-08 18:58:31 +00:00
serialization.fbs
Change the namespace name to avoid using
export
c++ keyword on namespace.
2024-12-20 16:02:15 -08: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 threefry with symbolic shapes
2025-01-07 09:10:04 +02:00
shape_poly.py
[shape_poly] Remove old non_negative support.
2025-01-06 07:36:11 -08:00
