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 Sep 13, 2024
None
Supported by