Implement Selective State Space Model (Mamba) in PyTorch
Hardmodern-architecturesauto-graded
Build Mamba's selective scan mechanism with input-dependent parameters, achieving linear-time sequence modeling without attention.
Solve it
Check your answer
The grader verifies properties of your implementation, so a correct solution written differently from ours still passes.
pip install torchleet
from torchleet import check
check("mamba", selective_scan, MambaBlock)