torch>=2.0.0
transformers>=4.40.0
mamba-ssm>=2.0.0
causal-conv1d>=1.2.0
einops>=0.7.0

[dev]
pytest>=7.0.0
pytest-asyncio>=0.21.0
