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, BEARCloud VMs, and CaStLeS VMs):

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

BEAR Apps Version

2020h

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