mamba

Classes

Mamba(*args, **kwargs)

(SN) Wraps a sequence of Mamba modules using the simple Mamba module from: https://github.com/state-spaces/mamba/blob/main/mamba_ssm/modules/mamba_simple.py