System Status: All systems are operational • Services are available and operational.
Click for detailed status

JAX

JAX is a Python library for accelerator-oriented array computation and program transformation, designed for high-performance numerical computing and large-scale machine learning.

Installation

To install JAX on Euler, you can use pip: https://pypi.org/project/jax/ To target nVidia GPUs, use the CUDA version. To target AMD GPUs, use the ROCm version. For CPU-only installations, you can use the CPU version.