torch>=2.1.2
timm==0.9.0
mamba-ssm==1.0.1
causal-conv1d==1.0.2
einops
