Dear users, we have installed a new software: jax 0.4.25-CUDA-12.1.1:
---------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------
jax: jax/0.4.25-CUDA-12.1.1
---------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------
Description:
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
You will need to load all module(s) on any one of the lines below before the "jax/0.4.25-CUDA-12.1.1" module is available to load.
GCC/12.3.0
Help:
Description
===========
Composable transformations of Python+NumPy programs:
differentiate, vectorize, JIT to GPU/TPU, and more
More information
================
- Homepage: https://pypi.python.org/pypi/jax
Included extensions
===================
jax-0.4.25
Best,
HPC team