.. diffmahnet documentation master file, created by
   sphinx-quickstart on Thu Apr 17 21:41:39 2025.
   You can adapt this file completely to your liking, but it should at least
   contain the root `toctree` directive.

Welcome to diffmahnet's documentation!
======================================

DiffMahNet is a python package for emulating the
`diffmah <https://diffmah.readthedocs.io>`__
model using normalizing flows powered by
`flowjax <https://danielward27.github.io/flowjax/>`__.
The code is open-source and available on
`GitHub <https://github.com/AlanPearl/diffmahnet>`__.

Overview
--------

* :doc:`installation`
* :doc:`reference`

Tutorials
---------

* :doc:`notebooks/intro`
* :doc:`notebooks/load_pretrained`


Indices and tables
------------------

* :ref:`genindex`

.. * :ref:`search`
.. * :ref:`modindex`

.. Hidden TOCs

.. toctree::
   :maxdepth: 2
   :caption: Overview
   :hidden:

   installation.rst
   reference.rst


.. toctree::
   :maxdepth: 2
   :caption: Tutorials
   :hidden:

   notebooks/intro.ipynb
   notebooks/load_pretrained.ipynb