| --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 | |