Avatar for JAX team from gravatar.com

JAX team

Username    jax-team
Date joined   Joined

10 projects

jaxlib

Last released

XLA library for JAX

jax-cuda13-plugin

Last released

JAX Plugin for NVIDIA GPUs

jax-cuda13-pjrt

Last released

JAX XLA PJRT Plugin for NVIDIA GPUs

jax-cuda12-plugin

Last released

JAX Plugin for NVIDIA GPUs

jax-cuda12-pjrt

Last released

JAX XLA PJRT Plugin for NVIDIA GPUs

jax

Last released

Differentiate, compile, and transform Numpy code.

mosaic-gpu-cuda12

Last released

An empty wheel for MOSAIC GPU CUDA12 reservation

mosaic-gpu-cuda13

Last released

An empty wheel for MOSAIC GPU CUDA13 reservation

jax-cuda11-plugin

Last released

JAX Plugin for NVIDIA GPUs

jax-cuda11-pjrt

Last released

JAX XLA PJRT Plugin for NVIDIA GPUs

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page