transformers
torch
accelerate
datasets
mamba_ssm
causal-conv1d
