requirements.txt

# This file was autogenerated by uv via the following command:
#    uv pip compile --all-extras pyproject.toml -o requirements.txt
aiohttp==3.9.3
    # via
    #   datasets
    #   fsspec
aiosignal==1.3.1
    # via aiohttp
antlr4-python3-runtime==4.9.3
    # via omegaconf
asttokens==2.4.1
    # via stack-data
attrs==23.2.0
    # via aiohttp
certifi==2024.2.2
    # via requests
charset-normalizer==3.3.2
    # via requests
contourpy==1.2.1
    # via matplotlib
coverage==7.4.4
    # via pytest-cov
cycler==0.12.1
    # via matplotlib
datasets==2.18.0
decorator==5.1.1
    # via
    #   ipdb
    #   ipython
dill==0.3.8
    # via
    #   datasets
    #   multiprocess
executing==2.0.1
    # via stack-data
filelock==3.13.1
    # via
    #   datasets
    #   huggingface-hub
    #   torch
    #   transformers
    #   triton
fonttools==4.50.0
    # via matplotlib
frozenlist==1.4.1
    # via
    #   aiohttp
    #   aiosignal
fsspec==2024.2.0
    # via
    #   datasets
    #   huggingface-hub
    #   lightning
    #   pytorch-lightning
    #   torch
greenlet==3.0.3
    # via pynvim
huggingface-hub==0.21.4
    # via
    #   datasets
    #   timm
    #   tokenizers
    #   transformers
idna==3.6
    # via
    #   requests
    #   yarl
iniconfig==2.0.0
    # via pytest
ipdb==0.13.13
ipython==8.22.2
    # via ipdb
jedi==0.19.1
    # via ipython
jinja2==3.1.3
    # via torch
joblib==1.3.2
    # via scikit-learn
kiwisolver==1.4.5
    # via matplotlib
lightning==2.2.1
lightning-utilities==0.11.0
    # via
    #   lightning
    #   pytorch-lightning
    #   torchmetrics
loguru==0.7.2
markupsafe==2.1.5
    # via jinja2
matplotlib==3.8.4
matplotlib-inline==0.1.6
    # via ipython
mpmath==1.3.0
    # via sympy
msgpack==1.0.8
    # via pynvim
multidict==6.0.5
    # via
    #   aiohttp
    #   yarl
multiprocess==0.70.16
    # via datasets
mypy==1.9.0
mypy-extensions==1.0.0
    # via mypy
networkx==3.2.1
    # via torch
numpy==1.26.4
    # via
    #   contourpy
    #   datasets
    #   lightning
    #   matplotlib
    #   pandas
    #   pyarrow
    #   pylibjpeg
    #   pylibjpeg-libjpeg
    #   pylibjpeg-openjpeg
    #   pytorch-lightning
    #   scikit-learn
    #   scipy
    #   torchmetrics
    #   torchvision
    #   transformers
nvidia-cublas-cu12==12.1.3.1
    # via
    #   nvidia-cudnn-cu12
    #   nvidia-cusolver-cu12
    #   torch
nvidia-cuda-cupti-cu12==12.1.105
    # via torch
nvidia-cuda-nvrtc-cu12==12.1.105
    # via torch
nvidia-cuda-runtime-cu12==12.1.105
    # via torch
nvidia-cudnn-cu12==8.9.2.26
    # via torch
nvidia-cufft-cu12==11.0.2.54
    # via torch
nvidia-curand-cu12==10.3.2.106
    # via torch
nvidia-cusolver-cu12==11.4.5.107
    # via torch
nvidia-cusparse-cu12==12.1.0.106
    # via
    #   nvidia-cusolver-cu12
    #   torch
nvidia-nccl-cu12==2.19.3
    # via torch
nvidia-nvjitlink-cu12==12.4.99
    # via
    #   nvidia-cusolver-cu12
    #   nvidia-cusparse-cu12
nvidia-nvtx-cu12==12.1.105
    # via torch
omegaconf==2.3.0
packaging==24.0
    # via
    #   datasets
    #   huggingface-hub
    #   lightning
    #   lightning-utilities
    #   matplotlib
    #   pytest
    #   pytorch-lightning
    #   torchmetrics
    #   transformers
pandas==2.2.1
    # via datasets
parso==0.8.3
    # via jedi
pexpect==4.9.0
    # via ipython
pillow==10.2.0
    # via
    #   matplotlib
    #   torchvision
pluggy==1.4.0
    # via pytest
prompt-toolkit==3.0.43
    # via ipython
ptyprocess==0.7.0
    # via pexpect
pure-eval==0.2.2
    # via stack-data
pyarrow==15.0.2
    # via datasets
pyarrow-hotfix==0.6
    # via datasets
pydicom==2.4.4
pygments==2.17.2
    # via ipython
pylibjpeg==2.0.0
pylibjpeg-libjpeg==2.0.2
pylibjpeg-openjpeg==2.2.0
pynvim==0.5.0
pyparsing==3.1.2
    # via matplotlib
pytest==8.1.1
    # via pytest-cov
pytest-cov==5.0.0
python-dateutil==2.9.0.post0
    # via
    #   matplotlib
    #   pandas
pytorch-lightning==2.2.1
    # via lightning
pytz==2024.1
    # via pandas
pyyaml==6.0.1
    # via
    #   datasets
    #   huggingface-hub
    #   lightning
    #   omegaconf
    #   pytorch-lightning
    #   timm
    #   transformers
regex==2023.12.25
    # via transformers
requests==2.31.0
    # via
    #   datasets
    #   huggingface-hub
    #   transformers
ruff==0.3.4
safetensors==0.4.2
    # via
    #   timm
    #   transformers
scikit-learn==1.4.1.post1
scipy==1.12.0
    # via scikit-learn
sentencepiece==0.2.0
setuptools==69.2.0
    # via lightning-utilities
six==1.16.0
    # via
    #   asttokens
    #   python-dateutil
stack-data==0.6.3
    # via ipython
sympy==1.12
    # via torch
threadpoolctl==3.4.0
    # via scikit-learn
timm==0.9.16
tokenizers==0.15.2
    # via transformers
torch==2.2.1
    # via
    #   lightning
    #   pytorch-lightning
    #   timm
    #   torchmetrics
    #   torchvision
torchmetrics==1.3.2
    # via
    #   lightning
    #   pytorch-lightning
torchvision==0.17.1
    # via timm
tqdm==4.66.2
    # via
    #   datasets
    #   huggingface-hub
    #   lightning
    #   pytorch-lightning
    #   transformers
traitlets==5.14.2
    # via
    #   ipython
    #   matplotlib-inline
transformers==4.39.1
triton==2.2.0
    # via torch
typing-extensions==4.10.0
    # via
    #   huggingface-hub
    #   lightning
    #   lightning-utilities
    #   mypy
    #   pytorch-lightning
    #   torch
tzdata==2024.1
    # via pandas
urllib3==2.2.1
    # via requests
wcwidth==0.2.13
    # via prompt-toolkit
xxhash==3.4.1
    # via datasets
yarl==1.9.4
    # via aiohttp