mamba_module.hpp File Reference

Mamba/S6 selective-state-space recurrence with the MambaLRP relevance rule. More...

#include <optional>
#include <initializer_list>
#include <vector>
#include "pulsatrix/module.hpp"

Include dependency graph for mamba_module.hpp:

Go to the source code of this file.

Classes
class	pulsatrix::MambaModule
	Core Mamba/S6 selective-scan recurrence (Gu & Dao 2023, arXiv:2312.00752), input (N, L, d_model) -> output (N, L, d_model). More...

Namespaces
namespace	pulsatrix

Detailed Description

Mamba/S6 selective-state-space recurrence with the MambaLRP relevance rule.