Installing JAX
==============

JAX is available to install via the `Python Package Index`_.
For full installation instructions, please refer to the `Install Guide`_ in the project README.

.. _Python Package Index: https://pypi.org/project/jax/
.. _Install Guide: https://github.com/google/jax#installation