Skip to main content
Ctrl+K

Developer Documentation#

JAX welcomes contributions from the community. See below for various install guides to get setup as a developer as well as developer-focused resources such as Jax Enhancement Proposals.

  • Contributing to JAX
  • Building from source
  • Internal APIs
  • Autodidax: JAX core from scratch
  • JAX Enhancement Proposals (JEPs)

previous

Generalized Convolutions in JAX

next

Contributing to JAX