Welcome to CATENets's documentation!
====================================

.. mdinclude:: ../README.md


API documentation
=================

JAX models
==========
.. toctree::
    :glob:
    :maxdepth: 2

    jax_models.rst

PyTorch models
==============
.. toctree::
    :glob:
    :maxdepth: 2

    torch_models.rst


Datasets
========
.. toctree::
    :glob:
    :maxdepth: 2

    datasets.rst
