JAX 0.1.77-fosscuda-2019b-Python-3.7.4

Deprecated: Use of this version of JAX is deprecated. More information on our Applications Support and Retention Policy.

JAX is Autograd and XLA, brought together for high-performance machine learning research.

Accessing JAX 0.1.77-fosscuda-2019b-Python-3.7.4

To load the module for JAX 0.1.77-fosscuda-2019b-Python-3.7.4 please use this command on the BEAR systems (BlueBEAR, BEAR Cloud VMs, and CaStLeS VMs):

module load JAX/0.1.77-fosscuda-2019b-Python-3.7.4

BEAR Apps Version


More Information

For more information visit the JAX website.


This version of JAX has a direct dependency on: Bazel/2.0.0-GCCcore-8.3.0 cuDNN/ fosscuda/2019b Python/3.7.4-GCCcore-8.3.0 SciPy-bundle/2019.10-fosscuda-2019b-Python-3.7.4

Last modified on 21st September 2020