jax-0.2.19
Composable transformations of Python+NumPy programs, differentiate, vectorize, JIT to GPU/TPU, and more
Project Homepage: jax
Usage:
module load jax/0.2.19-foss-2020b
Composable transformations of Python+NumPy programs, differentiate, vectorize, JIT to GPU/TPU, and more
Project Homepage: jax
Usage:
module load jax/0.2.19-foss-2020b