.. _extensions:

Extension guides

Guides for extending JAX's capabilities, and for building libraries that use or interface with JAX.

.. toctree:: :caption: Extensible JAX internals :maxdepth: 1

notebooks/Writing_custom_interpreters_in_Jax jax.extend

.. toctree:: :caption: Libraries and extensions :maxdepth: 1

building_on_jax