jax
080294cf - Load CUDA libraries up front with cdll.LoadLibrary().

Commit

200 days ago

Load CUDA libraries up front with cdll.LoadLibrary(). Works around a bug in CUDNN where it does not declare or locate its own NVRTC dependency. However, this approach should also improve our error messages in general when we load NVIDIA's libraries, so do it across the board for all libraries we use.

References

#29462 - Load CUDA libraries up front with cdll.LoadLibrary().

#477 - CI: 06/20/25 upstream sync

#31381 - Remove old ROCm build code

#31720 - Fix ann_test.py numerical bug in target reshape

#31768 - [ROCm] Support lowering through PJRT_Triton_Extension

#32115 - Relax version requirements for ROCm Jax Plugin wheels

#33157 - Resolve undefined behavior in bitshift unit test

#33186 - Make nvidia version data optional for ROCm builds

#579 - Create rocm-test-requirements.txt

#580 - GESVDJ support for ROCm GPUs in JAX

#581 - Fix/pallas tests shared memory