SSM
===

``tt.ssm`` contains state-space-style layers adapted to traceTorch's one-timestep recurrent interface.

Base Layer
----------

.. autoclass:: tracetorch.ssm.Layer
   :members:
   :member-order: bysource
   :show-inheritance:

Layers
------

.. autoclass:: tracetorch.ssm.S4
   :members:
   :member-order: bysource
   :show-inheritance:

.. autoclass:: tracetorch.ssm.S5
   :members:
   :member-order: bysource
   :show-inheritance:

.. autoclass:: tracetorch.ssm.S6
   :members:
   :member-order: bysource
   :show-inheritance:

.. autoclass:: tracetorch.ssm.Mamba
   :members:
   :member-order: bysource
   :show-inheritance: