requirements.txt

absl-py==2.1.0
asciitree==0.3.3
black==24.8.0
blessed==1.20.0
certifi==2022.12.7
charset-normalizer==2.1.1
click==8.1.7
configmypy==0.2.0
contextlib2==21.6.0
contourpy==1.3.0
cycler==0.12.1
docker-pycreds==0.4.0
exceptiongroup==1.2.2
fasteners==0.19
filelock==3.13.1
flow-vis==0.1
fonttools==4.54.1
fsspec==2024.2.0
gitdb==4.0.11
GitPython==3.1.43
gpustat==1.1.1
grpcio==1.66.1
h5py==3.11.0
huggingface-hub==0.25.1
idna==3.4
iniconfig==2.0.0
Jinja2==3.1.3
kiwisolver==1.4.7
Markdown==3.7
MarkupSafe==2.1.5
matplotlib==3.9.2
ml_collections==0.1.1
mpmath==1.3.0
mypy-extensions==1.0.0
networkx==3.2.1
neuraloperator==0.2.0
nibabel==5.2.1
numcodecs==0.13.0
numpy==1.26.3
nvidia-ml-py==12.560.30
opencv-python-headless==4.10.0.84
opt-einsum==3.3.0
packaging==24.1
pandas==2.2.3
pathspec==0.12.1
pillow==10.2.0
platformdirs==4.3.6
pluggy==1.5.0
prettytable==3.11.0
protobuf==5.28.2
psutil==6.0.0
pyparsing==3.1.4
pystrum==0.4
pytest==8.3.3
pytest-mock==3.14.0
python-dateutil==2.9.0.post0
pytz==2024.2
PyYAML==6.0.2
requests==2.28.1
ruamel.yaml==0.18.6
ruamel.yaml.clib==0.2.8
safetensors==0.4.5
scipy==1.14.1
sentry-sdk==2.14.0
setproctitle==1.3.3
six==1.16.0
smmap==5.0.1
sympy==1.12
tensorboard==2.17.1
tensorboard-data-server==0.7.2
tensorly==0.8.1
tensorly-torch==0.5.0
timm==1.0.9
tomli==2.0.1
torch==2.1.0+cu118
torch-harmonics==0.6.5
--extra-index-url https://download.pytorch.org/whl/cu118
torchaudio==2.1.0+cu118
torchvision==0.16.0+cu118
tqdm==4.66.5
triton==2.1.0
typing_extensions==4.9.0
tzdata==2024.2
urllib3==1.26.13
wandb==0.18.1
wcwidth==0.2.13
Werkzeug==3.0.4
zarr==2.18.3
PySocks==1.7.1 
beautifulsoup4==4.12.3 
gdown==5.2.0 
soupsieve==2.6