mirrors/rocm_jax
1
0
Fork 0
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_plugins
History
jax authors 787c79f2bb 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-04-09 18:05:12 +00:00
..
cuda
Update the minimum supported CuDNN version to 9.8 (previously 9.1).
2025-04-09 18:05:12 +00:00
rocm
[jaxlib:gpu] Clean up custom call GPU callback handling code.
2025-04-09 18:05:12 +00:00
BUILD.bazel
Don't pass --nocheck_visibility to Bazel.
2024-09-27 11:19:42 -04:00