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.

Visit the JAX website to create your account and explore pricing options.

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.