--extra-index-url https://download.pytorch.org/whl/cu121
torch==2.1.2
torchvision==0.16.2
torchaudio==2.1.2
numpy==1.26.2
pandas==2.1.3
PyYAML==6.0.1
nibabel==5.3.2
nilearn==0.10.3
scikit-learn==1.3.2
timm==0.4.12
einops==0.7.0
packaging>=23.2
tqdm==4.66.1
triton==2.1.0
causal-conv1d==1.2.2.post1
mamba-ssm==2.0.3
flash-attn==2.5.8