torch==2.2.2
flash-attn==2.5.9.post1
lightning==2.3.0
causal-conv1d==1.3.0.post1
mamba-ssm==2.1.0