.. _api-plugin-modes-training:

``pycif.plugins.modes.training`` — API reference
================================================

Configuration reference: :doc:`training plugin </documentation/plugins/modes/training>`

.. automodule:: pycif.plugins.modes.training.execute
   :members:
   :show-inheritance:
   :undoc-members:

.. automodule:: pycif.plugins.modes.training.trainer
   :members:
   :show-inheritance:
   :undoc-members:

.. automodule:: pycif.plugins.modes.training.backends.base
   :members:
   :show-inheritance:
   :undoc-members:

.. automodule:: pycif.plugins.modes.training.backends.jax_backend
   :members:
   :show-inheritance:
   :undoc-members:

.. automodule:: pycif.plugins.modes.training.backends.torch_backend
   :members:
   :show-inheritance:
   :undoc-members:

.. automodule:: pycif.plugins.modes.training.serialization.onnx_export
   :members:
   :show-inheritance:
   :undoc-members:

.. automodule:: pycif.plugins.modes.training.serialization.onnx_operator
   :members:
   :show-inheritance:
   :undoc-members:

.. automodule:: pycif.plugins.modes.training.tests.conftest
   :members:
   :show-inheritance:
   :undoc-members:

.. automodule:: pycif.plugins.modes.training.tests.test_adjoint
   :members:
   :show-inheritance:
   :undoc-members:

.. automodule:: pycif.plugins.modes.training.tests.test_onnx_roundtrip
   :members:
   :show-inheritance:
   :undoc-members: