.. multigrad documentation master file, created by
   sphinx-quickstart on Mon Jun 24 14:21:41 2024.
   You can adapt this file completely to your liking, but it should at least
   contain the root `toctree` directive.

Welcome to multigrad's documentation!
=====================================

multigrad allows you to easily implement data-parallelized (over MPI),
differentiable models in the Jax framework. The code is open-source and
available on `GitHub <https://github.com/AlanPearl/multigrad>`__.

.. toctree::
   :maxdepth: 2
   :caption: Contents:

   installation.rst
   notebooks/intro.ipynb
   reference.rst

Indices and tables
==================

* :ref:`genindex`
* :ref:`search`

.. * :ref:`modindex`