JAX
=====

.. toctree::
   :maxdepth: 2
   :glob:

   jax_plugin_api
   ../examples/frameworks/jax/index
   jax_fn