rocm.txt

# This file was autogenerated by uv via the following command:
#    uv pip compile requirements/test/rocm.in -c requirements/rocm.txt -o requirements/test/rocm.txt --index-strategy unsafe-best-match --python-platform x86_64-manylinux_2_28 --python-version 3.12 --no-emit-package torch --no-emit-package torchvision --no-emit-package torchaudio --no-emit-package triton --no-emit-package cuda-bindings --no-emit-package cuda-pathfinder --no-emit-package cuda-toolkit --no-emit-package cupy-cuda12x --no-emit-package nvidia-cublas --no-emit-package nvidia-cuda-cupti --no-emit-package nvidia-cuda-nvrtc --no-emit-package nvidia-cuda-runtime --no-emit-package nvidia-cudnn --no-emit-package nvidia-cufft --no-emit-package nvidia-cufile --no-emit-package nvidia-curand --no-emit-package nvidia-cusolver --no-emit-package nvidia-cusparse --no-emit-package nvidia-cusparselt --no-emit-package nvidia-nccl --no-emit-package nvidia-nvjitlink --no-emit-package nvidia-nvshmem --no-emit-package nvidia-nvtx --no-emit-package nvidia-cublas-cu12 --no-emit-package nvidia-cuda-cupti-cu12 --no-emit-package nvidia-cuda-nvrtc-cu12 --no-emit-package nvidia-cuda-runtime-cu12 --no-emit-package nvidia-cudnn-cu12 --no-emit-package nvidia-cufft-cu12 --no-emit-package nvidia-cufile-cu12 --no-emit-package nvidia-curand-cu12 --no-emit-package nvidia-cusolver-cu12 --no-emit-package nvidia-cusparse-cu12 --no-emit-package nvidia-cusparselt-cu12 --no-emit-package nvidia-nccl-cu12 --no-emit-package nvidia-nvjitlink-cu12 --no-emit-package nvidia-nvshmem-cu12 --no-emit-package nvidia-nvtx-cu12 --no-emit-package nvidia-cublas-cu13 --no-emit-package nvidia-cuda-cupti-cu13 --no-emit-package nvidia-cuda-nvrtc-cu13 --no-emit-package nvidia-cuda-runtime-cu13 --no-emit-package nvidia-cudnn-cu13 --no-emit-package nvidia-cufft-cu13 --no-emit-package nvidia-cufile-cu13 --no-emit-package nvidia-curand-cu13 --no-emit-package nvidia-cusolver-cu13 --no-emit-package nvidia-cusparse-cu13 --no-emit-package nvidia-cusparselt-cu13 --no-emit-package nvidia-nccl-cu13 --no-emit-package nvidia-nvjitlink-cu13 --no-emit-package nvidia-nvshmem-cu13 --no-emit-package nvidia-nvtx-cu13
absl-py==2.4.0
    # via
    #   rouge-score
    #   tensorboard
accelerate==1.13.0
    # via peft
aenum==3.1.17
    # via lightly
affine==2.4.0
    # via rasterio
aiohappyeyeballs==2.6.1
    # via aiohttp
aiohttp==3.13.3
    # via
    #   -c requirements/common.txt
    #   -r requirements/test/../common.txt
    #   aiohttp-cors
    #   fsspec
    #   gpt-oss
    #   lm-eval
    #   ray
aiohttp-cors==0.8.1
    # via ray
aiosignal==1.4.0
    # via aiohttp
albucore==0.1.2
    # via terratorch
albumentations==1.4.6
    # via
    #   -r requirements/test/rocm.in
    #   terratorch
alembic==1.18.4
    # via optuna
annotated-doc==0.0.4
    # via
    #   fastapi
    #   typer
annotated-types==0.7.0
    # via pydantic
anthropic==0.93.0
    # via
    #   -c requirements/common.txt
    #   -r requirements/test/../common.txt
antlr4-python3-runtime==4.9.3
    # via
    #   hydra-core
    #   omegaconf
anyio==4.13.0
    # via
    #   anthropic
    #   httpx
    #   mcp
    #   openai
    #   sse-starlette
    #   starlette
    #   watchfiles
arctic-inference==0.1.1
    # via -r requirements/test/rocm.in
argcomplete==3.6.3
    # via datamodel-code-generator
arrow==1.4.0
    # via isoduration
astor==0.8.1
    # via depyf
attrs==26.1.0
    # via
    #   aiohttp
    #   fiona
    #   jsonlines
    #   jsonschema
    #   pytest-subtests
    #   rasterio
    #   referencing
audioread==3.0.1
    # via librosa
av==16.1.0
    # via
    #   -r requirements/test/../common.txt
    #   -r requirements/test/rocm.in
azure-core==1.39.0
    # via
    #   azure-identity
    #   azure-storage-blob
azure-identity==1.25.3
    # via runai-model-streamer-azure
azure-storage-blob==12.28.0
    # via runai-model-streamer-azure
backoff==2.2.1
    # via
    #   -r requirements/test/rocm.in
    #   schemathesis
bitsandbytes==0.49.2
    # via
    #   -r requirements/test/rocm.in
    #   lightning
black==26.3.1
    # via datamodel-code-generator
blake3==1.0.8
    # via -r requirements/test/../common.txt
blobfile==3.0.0
    # via -r requirements/test/rocm.in
bm25s==0.2.13
    # via mteb
boto3==1.42.74
    # via
    #   runai-model-streamer-s3
    #   tensorizer
botocore==1.42.74
    # via
    #   boto3
    #   s3transfer
bounded-pool-executor==0.0.3
    # via pqdm
buildkite-test-collector==0.1.9
    # via -r requirements/test/rocm.in
cachetools==7.0.5
    # via -r requirements/test/../common.txt
cbor2==5.9.0
    # via -r requirements/test/../common.txt
certifi==2026.2.25
    # via
    #   fiona
    #   httpcore
    #   httpx
    #   lightly
    #   pyogrio
    #   pyproj
    #   rasterio
    #   requests
    #   sentry-sdk
cffi==1.17.1
    # via
    #   cryptography
    #   soundfile
chardet==5.2.0
    # via mbstrdecoder
charset-normalizer==3.4.6
    # via requests
choreographer==1.2.1
    # via kaleido
chz==0.4.0
    # via gpt-oss
click==8.3.1
    # via
    #   black
    #   click-plugins
    #   cligj
    #   fiona
    #   jiwer
    #   nltk
    #   rasterio
    #   ray
    #   rich-toolkit
    #   schemathesis
    #   typer
    #   uvicorn
    #   wandb
click-plugins==1.1.1.2
    # via fiona
cligj==0.7.2
    # via
    #   fiona
    #   rasterio
cloudpickle==3.1.2
    # via -r requirements/test/../common.txt
colorama==0.4.6
    # via
    #   perceptron
    #   sacrebleu
    #   schemathesis
colorful==0.5.8
    # via ray
colorlog==6.10.1
    # via optuna
compressed-tensors==0.15.0.1
    # via
    #   -c requirements/common.txt
    #   -r requirements/test/../common.txt
contourpy==1.3.3
    # via matplotlib
coverage==7.13.5
    # via pytest-cov
cramjam==2.11.0
    # via fastparquet
cryptography==46.0.0
    # via
    #   azure-identity
    #   azure-storage-blob
    #   google-auth
    #   msal
    #   pyjwt
cycler==0.12.1
    # via matplotlib
datamodel-code-generator==0.55.0
    # via -r requirements/test/rocm.in
dataproperty==1.1.0
    # via
    #   pytablewriter
    #   tabledata
datasets==3.6.0
    # via
    #   -r requirements/test/rocm.in
    #   evaluate
    #   lm-eval
    #   mteb
decorator==5.2.1
    # via librosa
decord==0.6.0
    # via -r requirements/test/rocm.in
depyf==0.20.0
    # via
    #   -c requirements/common.txt
    #   -r requirements/test/../common.txt
diffusers==0.37.0
    # via terratorch
dill==0.3.8
    # via
    #   datasets
    #   depyf
    #   evaluate
    #   lm-eval
    #   multiprocess
diskcache==5.6.3
    # via
    #   -c requirements/common.txt
    #   -r requirements/test/../common.txt
distlib==0.4.0
    # via virtualenv
distro==1.9.0
    # via
    #   anthropic
    #   openai
dnspython==2.8.0
    # via email-validator
docker==7.1.0
    # via gpt-oss
docopt==0.6.2
    # via num2words
docstring-parser==0.17.0
    # via
    #   anthropic
    #   jsonargparse
einops==0.8.2
    # via
    #   -r requirements/test/../common.txt
    #   -r requirements/test/rocm.in
    #   encodec
    #   terratorch
    #   torchgeo
    #   vector-quantize-pytorch
    #   vocos
einx==0.4.2
    # via vector-quantize-pytorch
email-validator==2.3.0
    # via
    #   fastapi
    #   pydantic
encodec==0.1.1
    # via vocos
et-xmlfile==2.0.0
    # via openpyxl
evaluate==0.4.6
    # via lm-eval
fastapi==0.135.2
    # via
    #   -c requirements/common.txt
    #   -r requirements/test/../common.txt
    #   gpt-oss
    #   model-hosting-container-standards
fastapi-cli==0.0.24
    # via fastapi
fastapi-cloud-cli==0.16.1
    # via fastapi-cli
fastar==0.10.0
    # via fastapi-cloud-cli
fastparquet==2026.3.0
    # via genai-perf
fastsafetensors @ git+https://github.com/foundation-model-stack/fastsafetensors.git@65d80088fca7a8f567fba30415fbcc80f7d2259c
    # via
    #   -c requirements/rocm.txt
    #   -r requirements/test/rocm.in
filelock==3.25.2
    # via
    #   -c requirements/common.txt
    #   -r requirements/test/../common.txt
    #   blobfile
    #   datasets
    #   diffusers
    #   huggingface-hub
    #   python-discovery
    #   ray
    #   torch
    #   virtualenv
fiona==1.10.1
    # via torchgeo
fonttools==4.62.1
    # via matplotlib
fqdn==1.5.1
    # via jsonschema
frozendict==2.4.7
    # via einx
frozenlist==1.8.0
    # via
    #   aiohttp
    #   aiosignal
fsspec==2025.3.0
    # via
    #   datasets
    #   evaluate
    #   fastparquet
    #   huggingface-hub
    #   lightning
    #   pytorch-lightning
    #   tacoreader
    #   torch
ftfy==6.3.1
    # via open-clip-torch
genai-perf==0.0.16
    # via -r requirements/test/rocm.in
genson==1.3.0
    # via datamodel-code-generator
geopandas==1.1.3
    # via terratorch
gguf==0.18.0
    # via
    #   -c requirements/common.txt
    #   -r requirements/test/../common.txt
gitdb==4.0.12
    # via gitpython
gitpython==3.1.46
    # via wandb
google-api-core==2.30.0
    # via
    #   google-cloud-core
    #   google-cloud-storage
    #   opencensus
google-auth==2.49.1
    # via
    #   google-api-core
    #   google-cloud-core
    #   google-cloud-storage
    #   runai-model-streamer-gcs
google-cloud-core==2.5.0
    # via google-cloud-storage
google-cloud-storage==3.10.1
    # via runai-model-streamer-gcs
google-crc32c==1.8.0
    # via
    #   google-cloud-storage
    #   google-resumable-media
google-resumable-media==2.8.0
    # via google-cloud-storage
googleapis-common-protos==1.73.0
    # via
    #   google-api-core
    #   opentelemetry-exporter-otlp-proto-grpc
    #   opentelemetry-exporter-otlp-proto-http
gpt-oss==0.0.8
    # via -r requirements/test/rocm.in
graphql-core==3.2.8
    # via hypothesis-graphql
greenlet==3.3.2
    # via sqlalchemy
grpcio==1.78.0
    # via
    #   -c requirements/rocm.txt
    #   -r requirements/test/rocm.in
    #   grpcio-reflection
    #   opentelemetry-exporter-otlp-proto-grpc
    #   ray
    #   tensorboard
grpcio-reflection==1.78.0
    # via
    #   -c requirements/rocm.txt
    #   -r requirements/test/rocm.in
h11==0.16.0
    # via
    #   httpcore
    #   uvicorn
h2==4.3.0
    # via httpx
h5py==3.16.0
    # via terratorch
harfile==0.4.0
    # via schemathesis
hf-xet==1.4.3
    # via huggingface-hub
hiredis==3.3.1
    # via tensorizer
hpack==4.1.0
    # via h2
html2text==2025.4.15
    # via gpt-oss
httpcore==1.0.9
    # via httpx
httptools==0.7.1
    # via uvicorn
httpx==0.27.2
    # via
    #   -r requirements/test/rocm.in
    #   anthropic
    #   diffusers
    #   fastapi
    #   fastapi-cloud-cli
    #   huggingface-hub
    #   mcp
    #   model-hosting-container-standards
    #   openai
    #   perceptron
    #   schemathesis
httpx-sse==0.4.3
    # via mcp
huggingface-hub==1.10.2
    # via
    #   accelerate
    #   datasets
    #   diffusers
    #   evaluate
    #   open-clip-torch
    #   peft
    #   segmentation-models-pytorch
    #   sentence-transformers
    #   terratorch
    #   timm
    #   tokenizers
    #   transformers
    #   vocos
humanize==4.15.0
    # via runai-model-streamer
hydra-core==1.3.2
    # via
    #   lightly
    #   lightning
hyperframe==6.1.0
    # via h2
hypothesis==6.151.9
    # via
    #   hypothesis-graphql
    #   hypothesis-jsonschema
    #   schemathesis
hypothesis-graphql==0.12.0
    # via schemathesis
hypothesis-jsonschema==0.23.1
    # via schemathesis
idna==3.11
    # via
    #   anyio
    #   email-validator
    #   httpx
    #   jsonschema
    #   requests
    #   yarl
ijson==3.5.0
    # via -r requirements/test/../common.txt
imagehash==4.3.2
    # via -r requirements/test/rocm.in
imageio==2.37.3
    # via scikit-image
importlib-metadata==8.7.1
    # via
    #   diffusers
    #   opentelemetry-api
importlib-resources==6.5.2
    # via typeshed-client
inflect==7.5.0
    # via datamodel-code-generator
iniconfig==2.3.0
    # via pytest
instanttensor==0.1.6
    # via -r requirements/test/rocm.in
interegular==0.3.3
    # via lm-format-enforcer
isodate==0.7.2
    # via azure-storage-blob
isoduration==20.11.0
    # via jsonschema
isort==8.0.1
    # via datamodel-code-generator
jinja2==3.1.6
    # via
    #   datamodel-code-generator
    #   fastapi
    #   genai-perf
    #   lm-eval
    #   torch
jiter==0.14.0
    # via
    #   anthropic
    #   openai
jiwer==4.0.0
    # via -r requirements/test/rocm.in
jmespath==1.1.0
    # via
    #   boto3
    #   botocore
    #   model-hosting-container-standards
joblib==1.5.3
    # via
    #   librosa
    #   nltk
    #   scikit-learn
jsonargparse==4.47.0
    # via
    #   lightning
    #   terratorch
jsonlines==4.0.0
    # via lm-eval
jsonnet==0.21.0
    # via jsonargparse
jsonpointer==3.1.0
    # via jsonschema
jsonschema==4.26.0
    # via
    #   hypothesis-jsonschema
    #   mcp
    #   mistral-common
    #   ray
    #   schemathesis
jsonschema-specifications==2025.9.1
    # via jsonschema
junit-xml==1.9
    # via schemathesis
kaldi-native-fbank==1.22.3
    # via -r requirements/test/rocm.in
kaleido==1.0.0
    # via genai-perf
kiwisolver==1.5.0
    # via matplotlib
kornia==0.8.2
    # via torchgeo
kornia-rs==0.1.10
    # via kornia
lark==1.2.2
    # via
    #   -c requirements/common.txt
    #   -r requirements/test/../common.txt
lazy-loader==0.4
    # via
    #   librosa
    #   scikit-image
libnacl==2.1.0
    # via tensorizer
librosa==0.10.2.post1
    # via -r requirements/test/rocm.in
lightly==1.5.22
    # via
    #   terratorch
    #   torchgeo
lightly-utils==0.0.2
    # via lightly
lightning==2.6.1
    # via
    #   terratorch
    #   torchgeo
lightning-utilities==0.15.3
    # via
    #   lightning
    #   pytorch-lightning
    #   torchmetrics
llguidance==1.3.0
    # via
    #   -c requirements/common.txt
    #   -r requirements/test/../common.txt
llvmlite==0.44.0
    # via numba
lm-eval==0.4.11
    # via -r requirements/test/rocm.in
lm-format-enforcer==0.11.3
    # via
    #   -c requirements/common.txt
    #   -r requirements/test/../common.txt
logistro==2.0.1
    # via
    #   choreographer
    #   kaleido
loguru==0.7.3
    # via compressed-tensors
lxml==6.0.2
    # via
    #   blobfile
    #   gpt-oss
    #   sacrebleu
mako==1.3.10
    # via alembic
markdown==3.10.2
    # via tensorboard
markdown-it-py==4.0.0
    # via rich
markupsafe==3.0.3
    # via
    #   jinja2
    #   mako
    #   werkzeug
matplotlib==3.10.8
    # via
    #   -r requirements/test/rocm.in
    #   lightning
    #   torchgeo
mbstrdecoder==1.1.4
    # via
    #   dataproperty
    #   pytablewriter
    #   typepy
mcp==1.27.0
    # via -r requirements/test/../common.txt
mdurl==0.1.2
    # via markdown-it-py
mistral-common==1.11.0
    # via
    #   -c requirements/common.txt
    #   -r requirements/test/../common.txt
    #   -r requirements/test/rocm.in
model-hosting-container-standards==0.1.14
    # via
    #   -c requirements/common.txt
    #   -r requirements/test/../common.txt
more-itertools==10.8.0
    # via
    #   inflect
    #   lm-eval
mpmath==1.3.0
    # via sympy
msal==1.35.1
    # via
    #   azure-identity
    #   msal-extensions
msal-extensions==1.3.1
    # via azure-identity
msgpack==1.1.2
    # via
    #   librosa
    #   ray
msgspec==0.21.0
    # via -r requirements/test/../common.txt
mteb==2.11.5
    # via -r requirements/test/rocm.in
multidict==6.7.1
    # via
    #   aiohttp
    #   yarl
multiprocess==0.70.16
    # via
    #   -r requirements/test/rocm.in
    #   datasets
    #   evaluate
mypy-extensions==1.1.0
    # via black
narwhals==2.18.0
    # via plotly
networkx==3.6.1
    # via
    #   scikit-image
    #   torch
ninja==1.13.0
    # via -r requirements/test/../common.txt
nltk==3.9.3
    # via rouge-score
num2words==0.5.14
    # via -r requirements/test/rocm.in
numba==0.61.2
    # via
    #   -c requirements/rocm.txt
    #   -r requirements/test/rocm.in
    #   librosa
numkong==7.1.1
    # via albucore
numpy==2.2.6
    # via
    #   -r requirements/test/../common.txt
    #   -r requirements/test/rocm.in
    #   accelerate
    #   albucore
    #   albumentations
    #   bitsandbytes
    #   bm25s
    #   contourpy
    #   cupy-cuda12x
    #   datasets
    #   decord
    #   diffusers
    #   einx
    #   encodec
    #   evaluate
    #   fastparquet
    #   genai-perf
    #   geopandas
    #   gguf
    #   h5py
    #   imagehash
    #   imageio
    #   librosa
    #   lightly
    #   lightly-utils
    #   lm-eval
    #   matplotlib
    #   mistral-common
    #   mteb
    #   numba
    #   opencv-python-headless
    #   optuna
    #   pandas
    #   patsy
    #   peft
    #   perceptron
    #   pycocotools
    #   pyogrio
    #   pytrec-eval-terrier
    #   pywavelets
    #   rasterio
    #   rioxarray
    #   rouge-score
    #   runai-model-streamer
    #   sacrebleu
    #   scikit-image
    #   scikit-learn
    #   scipy
    #   segmentation-models-pytorch
    #   sentence-transformers
    #   shapely
    #   soundfile
    #   soxr
    #   statsmodels
    #   tensorboard
    #   tensorboardx
    #   tensorizer
    #   terratorch
    #   tifffile
    #   torchgeo
    #   torchmetrics
    #   torchvision
    #   transformers
    #   tritonclient
    #   vocos
    #   xarray
    #   xgrammar
omegaconf==2.3.0
    # via
    #   hydra-core
    #   lightning
open-clip-torch==2.32.0
    # via -r requirements/test/rocm.in
openai==2.31.0
    # via
    #   -c requirements/common.txt
    #   -r requirements/test/../common.txt
openai-harmony==0.0.8
    # via
    #   -c requirements/common.txt
    #   -r requirements/test/../common.txt
    #   gpt-oss
opencensus==0.11.4
    # via ray
opencensus-context==0.1.3
    # via opencensus
opencv-python-headless==4.13.0.92
    # via
    #   -c requirements/common.txt
    #   -r requirements/test/../common.txt
    #   -r requirements/test/rocm.in
    #   albumentations
    #   mistral-common
openpyxl==3.1.5
    # via -r requirements/test/rocm.in
opentelemetry-api==1.40.0
    # via
    #   -c requirements/common.txt
    #   -r requirements/test/../common.txt
    #   opentelemetry-exporter-otlp-proto-grpc
    #   opentelemetry-exporter-otlp-proto-http
    #   opentelemetry-exporter-prometheus
    #   opentelemetry-sdk
    #   opentelemetry-semantic-conventions
opentelemetry-exporter-otlp==1.40.0
    # via
    #   -c requirements/common.txt
    #   -r requirements/test/../common.txt
opentelemetry-exporter-otlp-proto-common==1.40.0
    # via
    #   opentelemetry-exporter-otlp-proto-grpc
    #   opentelemetry-exporter-otlp-proto-http
opentelemetry-exporter-otlp-proto-grpc==1.40.0
    # via opentelemetry-exporter-otlp
opentelemetry-exporter-otlp-proto-http==1.40.0
    # via opentelemetry-exporter-otlp
opentelemetry-exporter-prometheus==0.61b0
    # via ray
opentelemetry-proto==1.40.0
    # via
    #   opentelemetry-exporter-otlp-proto-common
    #   opentelemetry-exporter-otlp-proto-grpc
    #   opentelemetry-exporter-otlp-proto-http
    #   ray
opentelemetry-sdk==1.40.0
    # via
    #   -c requirements/common.txt
    #   -r requirements/test/../common.txt
    #   opentelemetry-exporter-otlp-proto-grpc
    #   opentelemetry-exporter-otlp-proto-http
    #   opentelemetry-exporter-prometheus
    #   opentelemetry-semantic-conventions-ai
    #   ray
opentelemetry-semantic-conventions==0.61b0
    # via
    #   opentelemetry-sdk
    #   opentelemetry-semantic-conventions-ai
opentelemetry-semantic-conventions-ai==0.5.1
    # via
    #   -c requirements/common.txt
    #   -r requirements/test/../common.txt
optuna==3.6.1
    # via genai-perf
orjson==3.11.7
    # via
    #   genai-perf
    #   kaleido
outlines-core==0.2.11
    # via
    #   -c requirements/common.txt
    #   -r requirements/test/../common.txt
packaging==26.0
    # via
    #   -c requirements/rocm.txt
    #   accelerate
    #   bitsandbytes
    #   black
    #   datasets
    #   evaluate
    #   fastparquet
    #   geopandas
    #   huggingface-hub
    #   hydra-core
    #   kaleido
    #   kornia
    #   lazy-loader
    #   lightning
    #   lightning-utilities
    #   lm-format-enforcer
    #   matplotlib
    #   optuna
    #   peft
    #   plotly
    #   pooch
    #   pyogrio
    #   pytest
    #   pytest-rerunfailures
    #   pytorch-lightning
    #   ray
    #   rioxarray
    #   scikit-image
    #   statsmodels
    #   tensorboard
    #   tensorboardx
    #   torchmetrics
    #   transformers
    #   typepy
    #   wandb
    #   xarray
pandas==3.0.1
    # via
    #   datasets
    #   evaluate
    #   fastparquet
    #   genai-perf
    #   geopandas
    #   statsmodels
    #   tacoreader
    #   torchgeo
    #   xarray
partial-json-parser==0.2.1.1.post7
    # via -r requirements/test/../common.txt
pathspec==1.0.4
    # via black
pathvalidate==3.3.1
    # via pytablewriter
patsy==1.0.2
    # via statsmodels
peft==0.18.1
    # via -r requirements/test/rocm.in
perceptron==0.1.4
    # via -r requirements/test/rocm.in
perf-analyzer==0.1.0
    # via genai-perf
pillow==12.1.1
    # via
    #   -r requirements/test/../common.txt
    #   diffusers
    #   genai-perf
    #   imagehash
    #   imageio
    #   lightly-utils
    #   matplotlib
    #   mistral-common
    #   perceptron
    #   scikit-image
    #   segmentation-models-pytorch
    #   tensorboard
    #   torchgeo
    #   torchvision
platformdirs==4.3.6
    # via
    #   black
    #   pooch
    #   python-discovery
    #   virtualenv
    #   wandb
plotly==6.6.0
    # via
    #   -r requirements/test/rocm.in
    #   genai-perf
pluggy==1.6.0
    # via
    #   pytest
    #   pytest-cov
polars==1.39.3
    # via mteb
polars-runtime-32==1.39.3
    # via polars
pooch==1.8.2
    # via librosa
portalocker==3.2.0
    # via sacrebleu
pqdm==0.2.0
    # via -r requirements/test/rocm.in
prometheus-client==0.24.1
    # via
    #   -c requirements/common.txt
    #   -r requirements/test/../common.txt
    #   opentelemetry-exporter-prometheus
    #   prometheus-fastapi-instrumentator
    #   ray
prometheus-fastapi-instrumentator==7.1.0
    # via
    #   -c requirements/common.txt
    #   -r requirements/test/../common.txt
propcache==0.4.1
    # via
    #   aiohttp
    #   yarl
proto-plus==1.27.1
    # via google-api-core
protobuf==6.33.6
    # via
    #   -c requirements/common.txt
    #   -r requirements/test/../common.txt
    #   google-api-core
    #   googleapis-common-protos
    #   grpcio-reflection
    #   opentelemetry-proto
    #   proto-plus
    #   ray
    #   tensorboard
    #   tensorboardx
    #   tensorizer
    #   wandb
psutil==7.2.2
    # via
    #   -r requirements/test/../common.txt
    #   accelerate
    #   peft
    #   tensorizer
py==1.11.0
    # via pytest-forked
py-cpuinfo==9.0.0
    # via -r requirements/test/../common.txt
py-spy==0.4.1
    # via ray
pyarrow==23.0.1
    # via
    #   datasets
    #   genai-perf
    #   tacoreader
    #   terratorch
pyasn1==0.6.3
    # via pyasn1-modules
pyasn1-modules==0.4.2
    # via google-auth
pybase64==1.4.3
    # via -r requirements/test/../common.txt
pycocotools==2.0.11
    # via terratorch
pycountry==26.2.16
    # via pydantic-extra-types
pycparser==3.0
    # via cffi
pycryptodomex==3.23.0
    # via blobfile
pydantic==2.12.5
    # via
    #   -c requirements/common.txt
    #   -r requirements/test/../common.txt
    #   -r requirements/test/rocm.in
    #   albumentations
    #   anthropic
    #   compressed-tensors
    #   datamodel-code-generator
    #   fastapi
    #   fastapi-cloud-cli
    #   gpt-oss
    #   lightly
    #   lm-format-enforcer
    #   mcp
    #   mistral-common
    #   model-hosting-container-standards
    #   mteb
    #   openai
    #   openai-harmony
    #   pydantic-extra-types
    #   pydantic-settings
    #   ray
    #   wandb
    #   xgrammar
pydantic-core==2.41.5
    # via pydantic
pydantic-extra-types==2.11.1
    # via
    #   fastapi
    #   mistral-common
pydantic-settings==2.13.1
    # via
    #   fastapi
    #   mcp
pygments==2.19.2
    # via rich
pyjwt==2.12.1
    # via
    #   mcp
    #   msal
pyogrio==0.12.1
    # via geopandas
pyparsing==3.3.2
    # via
    #   matplotlib
    #   rasterio
pyproj==3.7.2
    # via
    #   geopandas
    #   rioxarray
    #   torchgeo
pyrate-limiter==3.9.0
    # via schemathesis
pystemmer==3.0.0
    # via mteb
pytablewriter==1.2.1
    # via lm-eval
pytest==8.3.5
    # via
    #   -r requirements/test/rocm.in
    #   buildkite-test-collector
    #   genai-perf
    #   pytest-asyncio
    #   pytest-cov
    #   pytest-forked
    #   pytest-mock
    #   pytest-rerunfailures
    #   pytest-shard
    #   pytest-subtests
    #   pytest-timeout
    #   schemathesis
pytest-asyncio==0.24.0
    # via -r requirements/test/rocm.in
pytest-cov==6.3.0
    # via -r requirements/test/rocm.in
pytest-forked==1.6.0
    # via -r requirements/test/rocm.in
pytest-mock==3.15.1
    # via genai-perf
pytest-rerunfailures==14.0
    # via -r requirements/test/rocm.in
pytest-shard==0.1.2
    # via -r requirements/test/rocm.in
pytest-subtests==0.14.2
    # via schemathesis
pytest-timeout==2.3.1
    # via -r requirements/test/rocm.in
python-box==7.4.1
    # via terratorch
python-dateutil==2.9.0.post0
    # via
    #   arrow
    #   botocore
    #   lightly
    #   matplotlib
    #   pandas
    #   typepy
python-discovery==1.2.0
    # via virtualenv
python-dotenv==1.2.2
    # via
    #   pydantic-settings
    #   uvicorn
python-json-logger==4.1.0
    # via -r requirements/test/../common.txt
python-multipart==0.0.26
    # via
    #   fastapi
    #   mcp
python-rapidjson==1.23
    # via tritonclient
pytokens==0.4.1
    # via black
pytorch-lightning==2.6.1
    # via
    #   lightly
    #   lightning
pytrec-eval-terrier==0.5.10
    # via mteb
pytz==2026.1.post1
    # via typepy
pywavelets==1.9.0
    # via imagehash
pyyaml==6.0.3
    # via
    #   -r requirements/test/../common.txt
    #   accelerate
    #   albumentations
    #   datamodel-code-generator
    #   datasets
    #   genai-perf
    #   gguf
    #   huggingface-hub
    #   jsonargparse
    #   lightning
    #   lm-format-enforcer
    #   omegaconf
    #   optuna
    #   peft
    #   pytorch-lightning
    #   ray
    #   responses
    #   schemathesis
    #   timm
    #   transformers
    #   uvicorn
    #   vocos
    #   wandb
pyzmq==27.1.0
    # via
    #   -c requirements/common.txt
    #   -r requirements/test/../common.txt
rapidfuzz==3.12.1
    # via
    #   -r requirements/test/rocm.in
    #   jiwer
rasterio==1.5.0
    # via
    #   rioxarray
    #   terratorch
    #   torchgeo
ray==2.54.0
    # via -r requirements/test/rocm.in
redis==7.3.0
    # via tensorizer
referencing==0.37.0
    # via
    #   jsonschema
    #   jsonschema-specifications
regex==2026.2.28
    # via
    #   -r requirements/test/../common.txt
    #   diffusers
    #   nltk
    #   open-clip-torch
    #   sacrebleu
    #   tiktoken
    #   transformers
requests==2.32.5
    # via
    #   -c requirements/common.txt
    #   -r requirements/test/../common.txt
    #   azure-core
    #   buildkite-test-collector
    #   datasets
    #   diffusers
    #   docker
    #   evaluate
    #   gguf
    #   google-api-core
    #   google-cloud-storage
    #   gpt-oss
    #   lightly
    #   lm-eval
    #   mistral-common
    #   msal
    #   mteb
    #   opentelemetry-exporter-otlp-proto-http
    #   pooch
    #   ray
    #   responses
    #   schemathesis
    #   starlette-testclient
    #   tacoreader
    #   tiktoken
    #   wandb
responses==0.26.0
    # via genai-perf
rfc3339-validator==0.1.4
    # via jsonschema
rfc3987==1.3.8
    # via jsonschema
rich==14.3.3
    # via
    #   genai-perf
    #   lightning
    #   mteb
    #   perceptron
    #   rich-toolkit
    #   terratorch
    #   typer
rich-toolkit==0.19.7
    # via
    #   fastapi-cli
    #   fastapi-cloud-cli
rignore==0.7.6
    # via fastapi-cloud-cli
rioxarray==0.22.0
    # via terratorch
rouge-score==0.1.2
    # via lm-eval
rpds-py==0.30.0
    # via
    #   jsonschema
    #   referencing
rtree==1.4.1
    # via torchgeo
runai-model-streamer==0.15.7
    # via
    #   -c requirements/rocm.txt
    #   -r requirements/test/rocm.in
runai-model-streamer-azure==0.15.7
    # via runai-model-streamer
runai-model-streamer-gcs==0.15.7
    # via runai-model-streamer
runai-model-streamer-s3==0.15.7
    # via runai-model-streamer
s3transfer==0.16.0
    # via boto3
sacrebleu==2.6.0
    # via lm-eval
safetensors==0.7.0
    # via
    #   accelerate
    #   diffusers
    #   open-clip-torch
    #   peft
    #   segmentation-models-pytorch
    #   timm
    #   transformers
schemathesis==3.39.15
    # via -r requirements/test/rocm.in
scikit-image==0.26.0
    # via
    #   albumentations
    #   terratorch
scikit-learn==1.8.0
    # via
    #   albumentations
    #   librosa
    #   lm-eval
    #   mteb
    #   sentence-transformers
    #   terratorch
scipy==1.17.1
    # via
    #   albumentations
    #   bm25s
    #   imagehash
    #   librosa
    #   mteb
    #   pytrec-eval-terrier
    #   scikit-image
    #   scikit-learn
    #   sentence-transformers
    #   statsmodels
    #   vocos
segmentation-models-pytorch==0.5.0
    # via
    #   -r requirements/test/rocm.in
    #   terratorch
    #   torchgeo
sentence-transformers==5.3.0
    # via
    #   -r requirements/test/rocm.in
    #   mteb
sentencepiece==0.2.1
    # via -r requirements/test/../common.txt
sentry-sdk==2.55.0
    # via
    #   fastapi-cloud-cli
    #   wandb
setproctitle==1.3.7
    # via -r requirements/test/../common.txt
setuptools==79.0.1
    # via
    #   -c requirements/common.txt
    #   -c requirements/rocm.txt
    #   -r requirements/test/../common.txt
    #   model-hosting-container-standards
    #   pytablewriter
    #   tensorboard
    #   torch
shapely==2.1.2
    # via
    #   geopandas
    #   torchgeo
shellingham==1.5.4
    # via
    #   perceptron
    #   typer
simplejson==3.20.2
    # via choreographer
six==1.17.0
    # via
    #   -c requirements/common.txt
    #   -r requirements/test/../common.txt
    #   junit-xml
    #   lightly
    #   opencensus
    #   python-dateutil
    #   rfc3339-validator
    #   rouge-score
smart-open==7.5.1
    # via ray
smmap==5.0.3
    # via gitdb
sniffio==1.3.1
    # via
    #   anthropic
    #   httpx
    #   openai
sortedcontainers==2.4.0
    # via hypothesis
soundfile==0.13.1
    # via
    #   -r requirements/test/../common.txt
    #   -r requirements/test/rocm.in
    #   genai-perf
    #   librosa
    #   mistral-common
soxr==0.5.0.post1
    # via
    #   librosa
    #   mistral-common
sqlalchemy==2.0.48
    # via
    #   alembic
    #   optuna
sqlitedict==2.1.0
    # via lm-eval
sse-starlette==3.3.4
    # via mcp
starlette==0.52.1
    # via
    #   fastapi
    #   mcp
    #   model-hosting-container-standards
    #   prometheus-fastapi-instrumentator
    #   schemathesis
    #   sse-starlette
    #   starlette-testclient
starlette-testclient==0.4.1
    # via schemathesis
statsmodels==0.14.6
    # via genai-perf
stringzilla==4.6.0
    # via albucore
structlog==25.5.0
    # via gpt-oss
supervisor==4.3.0
    # via model-hosting-container-standards
sympy==1.14.0
    # via
    #   einx
    #   torch
tabledata==1.3.4
    # via pytablewriter
tabulate==0.10.0
    # via sacrebleu
tacoreader==0.5.6
    # via terratorch
tblib==3.1.0
    # via -r requirements/test/rocm.in
tcolorpy==0.1.7
    # via pytablewriter
tenacity==9.1.4
    # via
    #   gpt-oss
    #   lm-eval
tensorboard==2.20.0
    # via terratorch
tensorboard-data-server==0.7.2
    # via tensorboard
tensorboardx==2.6.4
    # via lightning
tensorizer==2.10.1
    # via
    #   -c requirements/rocm.txt
    #   -r requirements/test/rocm.in
termcolor==3.3.0
    # via
    #   gpt-oss
    #   terratorch
terratorch==1.2.2
    # via -r requirements/test/rocm.in
threadpoolctl==3.6.0
    # via scikit-learn
tifffile==2026.3.3
    # via
    #   scikit-image
    #   terratorch
tiktoken==0.12.0
    # via
    #   -c requirements/common.txt
    #   -r requirements/test/../common.txt
    #   gpt-oss
    #   lm-eval
    #   mistral-common
timm==1.0.17
    # via
    #   -c requirements/rocm.txt
    #   -r requirements/test/rocm.in
    #   open-clip-torch
    #   segmentation-models-pytorch
    #   terratorch
    #   torchgeo
tokenizers==0.22.2
    # via
    #   -c requirements/common.txt
    #   -r requirements/test/../common.txt
    #   -r requirements/test/rocm.in
    #   transformers
tomli==2.4.0
    # via schemathesis
tomli-w==1.2.0
    # via schemathesis
torchgeo==0.7.0
    # via
    #   -r requirements/test/rocm.in
    #   terratorch
torchmetrics==1.9.0
    # via
    #   lightning
    #   pytorch-lightning
    #   terratorch
    #   torchgeo
tqdm==4.67.3
    # via
    #   -r requirements/test/../common.txt
    #   datasets
    #   evaluate
    #   gguf
    #   huggingface-hub
    #   lightly
    #   lightning
    #   lm-eval
    #   mteb
    #   nltk
    #   open-clip-torch
    #   openai
    #   optuna
    #   peft
    #   pqdm
    #   pytorch-lightning
    #   segmentation-models-pytorch
    #   sentence-transformers
    #   tacoreader
    #   terratorch
    #   transformers
transformers==5.5.3
    # via
    #   -c requirements/common.txt
    #   -r requirements/test/../common.txt
    #   -r requirements/test/rocm.in
    #   compressed-tensors
    #   genai-perf
    #   peft
    #   sentence-transformers
    #   transformers-stream-generator
    #   xgrammar
transformers-stream-generator==0.0.5
    # via -r requirements/test/rocm.in
tritonclient==2.66.0
    # via -r requirements/test/rocm.in
typeguard==4.5.1
    # via inflect
typepy==1.3.4
    # via
    #   dataproperty
    #   pytablewriter
    #   tabledata
typer==0.24.1
    # via
    #   fastapi-cli
    #   fastapi-cloud-cli
    #   fastsafetensors
    #   huggingface-hub
    #   perceptron
    #   transformers
typeshed-client==2.9.0
    # via jsonargparse
typing-extensions==4.15.0
    # via
    #   -c requirements/common.txt
    #   -r requirements/test/../common.txt
    #   aiosignal
    #   albumentations
    #   alembic
    #   anthropic
    #   anyio
    #   azure-core
    #   azure-identity
    #   azure-storage-blob
    #   chz
    #   fastapi
    #   grpcio
    #   huggingface-hub
    #   librosa
    #   lightning
    #   lightning-utilities
    #   lm-eval
    #   mcp
    #   mistral-common
    #   mteb
    #   openai
    #   opentelemetry-api
    #   opentelemetry-exporter-otlp-proto-grpc
    #   opentelemetry-exporter-otlp-proto-http
    #   opentelemetry-sdk
    #   opentelemetry-semantic-conventions
    #   pqdm
    #   pydantic
    #   pydantic-core
    #   pydantic-extra-types
    #   pytorch-lightning
    #   referencing
    #   rich-toolkit
    #   sentence-transformers
    #   sqlalchemy
    #   starlette
    #   torch
    #   torchgeo
    #   typeguard
    #   typeshed-client
    #   typing-inspection
    #   wandb
    #   xgrammar
typing-inspection==0.4.2
    # via
    #   fastapi
    #   mcp
    #   pydantic
    #   pydantic-settings
tzdata==2025.3
    # via arrow
uri-template==1.3.0
    # via jsonschema
urllib3==2.6.3
    # via
    #   blobfile
    #   botocore
    #   docker
    #   lightly
    #   requests
    #   responses
    #   sentry-sdk
    #   tritonclient
uvicorn==0.42.0
    # via
    #   fastapi
    #   fastapi-cli
    #   fastapi-cloud-cli
    #   gpt-oss
    #   mcp
uvloop==0.22.1
    # via uvicorn
vector-quantize-pytorch==1.28.0
    # via -r requirements/test/rocm.in
virtualenv==21.2.0
    # via ray
vocos==0.1.0
    # via -r requirements/test/rocm.in
wandb==0.25.1
    # via terratorch
watchfiles==1.1.1
    # via
    #   -r requirements/test/../common.txt
    #   uvicorn
wcwidth==0.6.0
    # via ftfy
webcolors==25.10.0
    # via jsonschema
websockets==16.0
    # via uvicorn
werkzeug==3.1.6
    # via
    #   schemathesis
    #   tensorboard
word2number==1.1
    # via lm-eval
wrapt==2.1.2
    # via smart-open
xarray==2026.2.0
    # via rioxarray
xgrammar==0.1.33
    # via
    #   -c requirements/common.txt
    #   -r requirements/test/../common.txt
xxhash==3.6.0
    # via
    #   datasets
    #   evaluate
yarl==1.23.0
    # via
    #   aiohttp
    #   schemathesis
zipp==3.23.0
    # via importlib-metadata
zstandard==0.25.0
    # via lm-eval

# The following packages were excluded from the output:
# torch
# torchvision
# torchaudio
# triton
# cuda-bindings
# cuda-pathfinder
# cuda-toolkit
# cupy-cuda12x
# nvidia-cublas
# nvidia-cuda-cupti
# nvidia-cuda-nvrtc
# nvidia-cuda-runtime
# nvidia-cufft
# nvidia-cufile
# nvidia-curand
# nvidia-cusolver
# nvidia-cusparse
# nvidia-nvjitlink
# nvidia-nvtx
# nvidia-cudnn-cu13
# nvidia-cusparselt-cu13
# nvidia-nccl-cu13
# nvidia-nvshmem-cu13