Avatar for JAX team from gravatar.com

JAX team

Username    jax-team
Date joined   Joined

6 projects

jax-cuda12-plugin

Last released

JAX Plugin for NVIDIA GPUs

jax-cuda12-pjrt

Last released

JAX XLA PJRT Plugin for NVIDIA GPUs

jaxlib

Last released

XLA library for JAX

jax

Last released

Differentiate, compile, and transform Numpy code.

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 AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page