API Reference
=============

Complete reference documentation for all models, losses, and utilities.

.. toctree::
   :maxdepth: 2

   models
   losses
   stochastic