JAX 0.1.77-fosscuda-2019b-Python-3.7.4
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.
Dependencies
This version of JAX has a direct dependency on: Bazel/2.0.0-GCCcore-8.3.0 cuDNN/7.6.4.38-gcccuda-2019b 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