Dear users, we have installed a new software: jax 0.3.25-CUDA-11.7.0:
-----------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------
jax: jax/0.3.25-CUDA-11.7.0
-----------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------
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.3.25-CUDA-11.7.0" module is available to load.
GCC/11.3.0 OpenMPI/4.1.4
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
===================
etils-0.8.0, jax-0.3.25, opt_einsum-3.3.0
Best,
HPC team