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

.. toctree::
   :maxdepth: 2
   :caption: API documentation

   api/core
   api/torch_ckpt