7 projects
jaxlib
XLA library for JAX
jax-cuda13-plugin
JAX Plugin for NVIDIA GPUs
jax-cuda13-pjrt
JAX XLA PJRT Plugin for NVIDIA GPUs
jax-cuda12-plugin
JAX Plugin for NVIDIA GPUs
jax-cuda12-pjrt
JAX XLA PJRT Plugin for NVIDIA GPUs
jax
Differentiate, compile, and transform Numpy code.
ml-dtypes
ml_dtypes is a stand-alone implementation of several NumPy dtype extensions used in machine learning.