Official Docs Intermediate

JAX Official Documentation

Google

LevelIntermediate
Who it's forDevelopers who need high-performance numerical computing and automatic differentiation
LanguageEnglish
PriceFree
TypeOfficial Docs
CategoryDeep Learning
Link check2026-10-18

Overview

Documentation for Google's accelerated numerical computing library, supporting automatic differentiation and GPU/TPU compilation — a popular choice for research and high-performance experiments.

JAXGoogle
Visit resource ↗

https://docs.jax.dev/en/latest/