.. Quantax documentation master file, created by
   sphinx-quickstart on Thu Aug  1 15:06:03 2024.
   You can adapt this file completely to your liking, but it should at least
   contain the root `toctree` directive.

Quantax documentation
=====================

**Quantax** is a Python package for flexible neural quantum states based on 
`QuSpin <https://github.com/QuSpin/QuSpin/tree/dev_0.3.8>`_, 
`JAX <https://github.com/google/jax>`_, and 
`Equinox <https://github.com/patrick-kidger/equinox>`_.

.. toctree::
   :maxdepth: 1

   installation

.. toctree::
   :maxdepth: 2
   
   api