Last released Oct 22, 2024
JAX Plugin for NVIDIA GPUs
JAX XLA PJRT Plugin for NVIDIA GPUs
XLA library for JAX
Differentiate, compile, and transform Numpy code.
Last released Feb 26, 2024
Supported by