mamba_block

Classes

MambaBlock(*args, **kwargs)

Wraps the original Mamba module from, with added functionality for bidirectional processing: https://github.com/state-spaces/mamba/blob/main/mamba_ssm/modules/mamba_simple.py