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-16 11:56:07 +00:00
Code
Issues
Packages
Projects
Releases
Wiki
Activity
rocm_jax
/
jax
/
_src
/
lib
History
Michael Hudgins
d4d1518c3d
Update references to the GitHub url in JAX codebase to reflect move from google/jax to jax-ml/jax
...
PiperOrigin-RevId: 676843138
2024-09-20 07:52:33 -07:00
..
mlir
Partially rolling forward
#22998
2024-09-02 04:44:47 -07:00
__init__.py
Update references to the GitHub url in JAX codebase to reflect move from google/jax to jax-ml/jax
2024-09-20 07:52:33 -07:00
BUILD
Clean up BUILD files.
2024-08-26 09:11:17 -07:00
mosaic_gpu.py
Make CuDNN SDPA API work with JAX with a CUDA plugin configuration.
2024-06-06 12:09:19 -04:00
triton.py
Do not import from lowering in tests/pallas/pallas_test.py
2024-05-10 14:25:17 -07:00
