torch
=====

.. toctree::
   :maxdepth: 1

   torch_models

config
------
.. automodule:: flatiron.torch.config
    :members:
    :private-members:
    :undoc-members:
    :show-inheritance:

loss
----
.. automodule:: flatiron.torch.loss
    :members:
    :private-members:
    :undoc-members:
    :show-inheritance:

metric
------
.. automodule:: flatiron.torch.metric
    :members:
    :private-members:
    :undoc-members:
    :show-inheritance:

optimizer
---------
.. automodule:: flatiron.torch.optimizer
    :members:
    :private-members:
    :undoc-members:
    :show-inheritance:

tools
-----
.. automodule:: flatiron.torch.tools
    :members:
    :private-members:
    :undoc-members:
    :show-inheritance: