API Reference
===============
Welcome to gato-hep's documentation! The following modules are contained in the API reference: 

.. toctree::
   :maxdepth: 1

   models
   losses
   utils
   plotting_utils
   data_generation