torch>=2.2
pytorch-lightning
packaging
wandb
numpy
pandas
biopython
jsonargparse
transformers
tqdm
wheel
causal-conv1d
mamba-ssm
