.. automodule:: max.nn.state_space
   :no-members:

.. currentmodule:: max.nn.state_space

.. autosummary::
   :nosignatures:

   max.nn.state_space.kimi_delta

.. toctree::
   :maxdepth: 1
   :hidden:

   nn.state_space.kimi_delta

.. autosummary::
   :nosignatures:
   :toctree: generated
   :template: autosummary/function.rst

   causal_conv1d_varlen_fwd
   gated_delta_conv1d_fwd
   gated_delta_conv1d_verify_fwd
   gated_delta_recurrence_fwd
   gated_delta_recurrence_verify_ring_fwd
   gated_delta_state_fold
   gated_group_rmsnorm
   kda_chunk
   kda_chunk_supports_head_dims
   kda_decode
   mamba2_ssd_chunk_scan_varlen_fwd_inplace
   short_conv_ring
   verify_width_operand