mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-16 03:46:06 +00:00
Code Issues Packages Projects Releases Wiki Activity
rocm_jax/jax_plugins
History
jax authors e342f2dd60 Update the minimum supported CuDNN version to 9.8 (previously 9.1). 
Announce maximum supported CUDA version 12.8 (previously 12.3).

PiperOrigin-RevId: 741188737
2025-03-27 09:54:00 -07:00
..
cuda
Update the minimum supported CuDNN version to 9.8 (previously 9.1).
2025-03-27 09:54:00 -07:00
rocm
[jaxlib:gpu] Clean up custom call GPU callback handling code.
2025-03-25 11:43:10 -07:00
BUILD.bazel
Don't pass --nocheck_visibility to Bazel.
2024-09-27 11:19:42 -04:00