13 projects
jax-cuda12-pjrt
JAX XLA PJRT Plugin for NVIDIA GPUs
jax
Differentiate, compile, and transform Numpy code.
jax-cuda12-plugin
JAX Plugin for NVIDIA GPUs
ai-edge-litert-nightly
LiteRT is for mobile and embedded devices.
jax-cuda13-plugin
JAX Plugin for NVIDIA GPUs
xla-cpu-pjrt
XLA PJRT Plugin
ai-edge-litert
LiteRT is for mobile and embedded devices.
libtpu
Google Cloud TPU runtime library.
xla-cuda12-pjrt
XLA PJRT Plugin
litert-converter
LiteRT is for mobile and embedded devices.
jax-cuda13-pjrt
JAX XLA PJRT Plugin for NVIDIA GPUs
xla-cuda13-pjrt
XLA PJRT Plugin
jaxlib
XLA library for JAX
