Networks
========

Core interfaces and layers for building PPO networks.

Types and Interfaces
--------------------

.. automodule:: nnx_ppo.networks.types
   :members:
   :special-members: __call__, __init__
   :show-inheritance:

Containers
----------

.. automodule:: nnx_ppo.networks.containers
   :members:
   :special-members: __call__, __init__
   :show-inheritance:

Utilities
---------

.. automodule:: nnx_ppo.networks.utils
   :members:
   :special-members: __call__, __init__
   :show-inheritance:

Action Sampling
---------------

.. automodule:: nnx_ppo.networks.sampling_layers
   :members:
   :special-members: __call__, __init__
   :show-inheritance:

PPO Adapter
-----------

.. automodule:: nnx_ppo.networks.adapter
   :members:
   :special-members: __call__, __init__
   :show-inheritance:

Feedforward Layers
------------------

.. automodule:: nnx_ppo.networks.feedforward
   :members:
   :special-members: __call__, __init__
   :show-inheritance:

Recurrent Layers
----------------

.. automodule:: nnx_ppo.networks.recurrent
   :members:
   :special-members: __call__, __init__
   :show-inheritance:

Delays
------

.. automodule:: nnx_ppo.networks.delay
   :members:
   :special-members: __call__, __init__
   :show-inheritance:

Population Graphs
-----------------

.. automodule:: nnx_ppo.networks.graph
   :members:
   :special-members: __call__, __init__
   :show-inheritance:

.. automodule:: nnx_ppo.networks.graph.graph
   :members:
   :special-members: __call__, __init__
   :show-inheritance:

Variational Layers
------------------

.. automodule:: nnx_ppo.networks.variational
   :members:
   :special-members: __call__, __init__
   :show-inheritance:

Observation Normalization
-------------------------

.. automodule:: nnx_ppo.networks.normalizer
   :members:
   :special-members: __call__, __init__
   :show-inheritance:

Factory Functions
-----------------

.. automodule:: nnx_ppo.networks.factories
   :members:
   :show-inheritance:

Activation Recording
--------------------

.. automodule:: nnx_ppo.networks.recording
   :members:
   :special-members: __call__, __init__
   :show-inheritance: