Get Started with JAX

High-performance machine learning research with Autograd and XLA integration.

Getting Started

The JAX team maintains comprehensive docs that cover installation, configuration, and common patterns.

JAX offers a free tier — sign up to get started without any payment.

Our full tool profile covers JAX's strengths, weaknesses, pricing, and how it compares to alternatives.

Researchers who need both speed and flexibility in their ML workflows.

Teams working on complex models that require hardware acceleration.

Developers looking to integrate automatic differentiation into their projects.