Jaxlib with CUDA support
Before using the library, you have to load it:
module load profile/deeplrn
module load jaxlib
More detailed instructions on how to link the library to you code are in the help of the module
module help jaxlib
CINECA consultants can be reached through the address: superc@cineca.it