Algorithms
==========

Core training algorithm, rollout logic, configuration, and utilities.

Types
-----

.. automodule:: nnx_ppo.algorithms.types
   :members:
   :show-inheritance:

Configuration
-------------

.. automodule:: nnx_ppo.algorithms.config
   :members:
   :show-inheritance:

Training
--------

.. automodule:: nnx_ppo.algorithms.ppo
   :members:
   :show-inheritance:

Rollout
-------

.. automodule:: nnx_ppo.algorithms.rollout
   :members:
   :show-inheritance:

Checkpointing
-------------

.. automodule:: nnx_ppo.algorithms.checkpointing
   :members:
   :special-members: __call__
   :show-inheritance:

Callbacks
---------

.. automodule:: nnx_ppo.algorithms.callbacks
   :members:
   :show-inheritance:

Metrics
-------

.. automodule:: nnx_ppo.algorithms.metrics
   :members:
   :show-inheritance: