Developer notes

JAX welcomes contributions from the community. These are guides to get set up as a developer, as well as developer-focused resources, such as JAX Enhancement Proposals.

See also the extension guides, which document some of JAX’s (extensible) internals.

Contribution guides

• Contributing to JAX
• Building from source
• Investigating a regression

Design and internals

• Autodidax: JAX core from scratch
• JAX Enhancement Proposals (JEPs)
• Internal API reference