jax/0.1.39

name
JAX
version
0.1.39 compiled with gcc/6.2
description
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
url
https://github.com/google/jax
license
opensource
built
Thurs July 18 14:13:10 CDT 2019
tags
deep-learning, differentiation, numerical analysis
usage
Use the module system to load this version of jax: module load midway2; module load jax/0.1.39

See also

jax/0.1.39 (default)
All versions available for jax.
Software Modules
Full list of software modules available on Midway.
Software Modules Tutorial
A tutorial on Midway modules and how to use them.