.. _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
   Custom_Operation_for_GPUs
   jax.extend

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

   building_on_jax