JAXJAX is Autograd and XLA, brought together for high-performance machine learning research.
For more information visit the JAX website.
These versions of JAX are available on the BEAR systems (BlueBEAR, BEARCloud VMs, and CaStLeS VMs). These will be retained in accordance with our Applications Support and Retention Policy.
|Version||BEAR Apps Version|