Getting Started
Reference Documentation
Advanced JAX Tutorials
Notes
custom_vjp
nondiff_argnums
Developer documentation
API documentation
Warning
This page was created from a pull request (#9655).
previous
Autodidax: JAX core from scratch
next
Custom JVP/VJP rules for JAX-transformable functions