main
txt 1,457 lines 32.6 KB
Raw
1 # This file was autogenerated by uv via the following command:
2 # 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
3 absl-py==2.4.0
4 # via rouge-score
5 accelerate==1.13.0
6 # via peft
7 aiohappyeyeballs==2.6.1
8 # via aiohttp
9 aiohttp==3.14.3
10 # via
11 # -c requirements/common.txt
12 # -r requirements/test/../common.txt
13 # aiohttp-cors
14 # fsspec
15 # gpt-oss
16 # lm-eval
17 # ray
18 aiohttp-cors==0.8.1
19 # via ray
20 aiosignal==1.4.0
21 # via aiohttp
22 albumentations==1.4.6
23 # via -r requirements/test/rocm.in
24 alembic==1.18.4
25 # via optuna
26 annotated-doc==0.0.4
27 # via
28 # fastapi
29 # typer
30 annotated-types==0.7.0
31 # via pydantic
32 anthropic==0.93.0
33 # via
34 # -c requirements/common.txt
35 # -r requirements/test/../common.txt
36 anyio==4.14.2
37 # via
38 # -r requirements/test/rocm.in
39 # anthropic
40 # httpx
41 # httpx2
42 # mcp
43 # openai
44 # sse-starlette
45 # starlette
46 # watchfiles
47 apache-tvm-ffi==0.1.11
48 # via
49 # -c requirements/rocm.txt
50 # tilelang
51 # xgrammar
52 arctic-inference==0.1.1
53 # via -r requirements/test/rocm.in
54 argcomplete==3.6.3
55 # via datamodel-code-generator
56 astor==0.8.1
57 # via depyf
58 attrs==26.1.0
59 # via
60 # aiohttp
61 # jsonschema
62 # referencing
63 audioread==3.0.1
64 # via librosa
65 av==16.1.0
66 # via -r requirements/test/rocm.in
67 azure-core==1.39.0
68 # via
69 # azure-identity
70 # azure-storage-blob
71 azure-identity==1.25.3
72 # via runai-model-streamer-azure
73 azure-storage-blob==12.28.0
74 # via runai-model-streamer-azure
75 backoff==2.2.1
76 # via -r requirements/test/rocm.in
77 bitsandbytes==0.50.0
78 # via -r requirements/test/rocm.in
79 black==26.3.1
80 # via
81 # -r requirements/test/rocm.in
82 # datamodel-code-generator
83 blake3==1.0.8
84 # via -r requirements/test/../common.txt
85 blobfile==3.0.0
86 # via
87 # -r requirements/test/rocm.in
88 # gpt-oss
89 bm25s==0.2.13
90 # via mteb
91 boto3==1.42.74
92 # via
93 # runai-model-streamer-s3
94 # tensorizer
95 botocore==1.42.74
96 # via
97 # boto3
98 # s3transfer
99 bounded-pool-executor==0.0.3
100 # via pqdm
101 buildkite-test-collector==0.1.9
102 # via -r requirements/test/rocm.in
103 cachetools==7.0.5
104 # via -r requirements/test/../common.txt
105 cbor2==5.9.0
106 # via -r requirements/test/../common.txt
107 certifi==2026.2.25
108 # via
109 # httpcore
110 # httpx
111 # requests
112 # sentry-sdk
113 cffi==2.1.1
114 # via
115 # cryptography
116 # soundfile
117 chardet==5.2.0
118 # via mbstrdecoder
119 charset-normalizer==3.4.6
120 # via requests
121 choreographer==1.2.1
122 # via kaleido
123 chz==0.4.0
124 # via gpt-oss
125 click==8.4.2
126 # via
127 # black
128 # huggingface-hub
129 # jiwer
130 # nltk
131 # ray
132 # rich-toolkit
133 # schemathesis
134 # typer
135 # uvicorn
136 cloudpickle==3.1.2
137 # via
138 # -r requirements/test/../common.txt
139 # tilelang
140 cohere==7.0.8
141 # via -r requirements/test/rocm.in
142 cohere-melody==0.14.0
143 # via -r requirements/test/rocm.in
144 colorama==0.4.6
145 # via
146 # perceptron
147 # sacrebleu
148 colorful==0.5.8
149 # via ray
150 colorlog==6.10.1
151 # via optuna
152 compressed-tensors==0.17.0
153 # via
154 # -c requirements/common.txt
155 # -r requirements/test/../common.txt
156 contourpy==1.3.3
157 # via matplotlib
158 coverage==7.13.5
159 # via pytest-cov
160 cramjam==2.11.0
161 # via fastparquet
162 cryptography==50.0.1
163 # via
164 # -r requirements/test/rocm.in
165 # azure-identity
166 # azure-storage-blob
167 # google-auth
168 # msal
169 # pyjwt
170 cycler==0.12.1
171 # via matplotlib
172 datamodel-code-generator==0.79.0
173 # via -r requirements/test/rocm.in
174 dataproperty==1.1.0
175 # via
176 # pytablewriter
177 # tabledata
178 datasets==3.6.0
179 # via
180 # -r requirements/test/rocm.in
181 # evaluate
182 # lm-eval
183 # mteb
184 decorator==5.2.1
185 # via librosa
186 decord==0.6.0
187 # via -r requirements/test/rocm.in
188 defusedxml==0.7.1
189 # via nltk
190 depyf==0.20.0
191 # via
192 # -c requirements/common.txt
193 # -r requirements/test/../common.txt
194 dill==0.3.8
195 # via
196 # datasets
197 # depyf
198 # evaluate
199 # lm-eval
200 # multiprocess
201 distlib==0.4.0
202 # via virtualenv
203 distro==1.9.0
204 # via
205 # anthropic
206 # openai
207 dnspython==2.8.0
208 # via email-validator
209 docker==7.2.0
210 # via gpt-oss
211 docopt==0.6.2
212 # via num2words
213 docstring-parser==0.17.0
214 # via anthropic
215 einops==0.8.2
216 # via
217 # -r requirements/test/../common.txt
218 # encodec
219 # vector-quantize-pytorch
220 # vocos
221 einx==0.4.2
222 # via vector-quantize-pytorch
223 email-validator==2.3.0
224 # via
225 # fastapi
226 # pydantic
227 encodec==0.1.1
228 # via vocos
229 et-xmlfile==2.0.0
230 # via openpyxl
231 evaluate==0.4.6
232 # via lm-eval
233 fastapi==0.135.2
234 # via
235 # -c requirements/common.txt
236 # -r requirements/test/../common.txt
237 # gpt-oss
238 # model-hosting-container-standards
239 fastapi-cli==0.0.24
240 # via fastapi
241 fastapi-cloud-cli==0.16.1
242 # via fastapi-cli
243 fastar==0.10.0
244 # via fastapi-cloud-cli
245 fastavro==1.12.2
246 # via cohere
247 fastparquet==2026.3.0
248 # via genai-perf
249 fastsafetensors==0.3.3
250 # via
251 # -c requirements/rocm.txt
252 # -r requirements/test/rocm.in
253 filelock==3.25.2
254 # via
255 # -c requirements/common.txt
256 # -r requirements/test/../common.txt
257 # blobfile
258 # datasets
259 # huggingface-hub
260 # python-discovery
261 # ray
262 # torch
263 # virtualenv
264 fonttools==4.62.1
265 # via
266 # -r requirements/test/rocm.in
267 # matplotlib
268 frozendict==2.4.7
269 # via einx
270 frozenlist==1.8.0
271 # via
272 # aiohttp
273 # aiosignal
274 fsspec==2025.3.0
275 # via
276 # datasets
277 # evaluate
278 # fastparquet
279 # huggingface-hub
280 # torch
281 ftfy==6.3.1
282 # via open-clip-torch
283 genai-perf==0.0.16
284 # via -r requirements/test/rocm.in
285 genson==1.3.0
286 # via datamodel-code-generator
287 google-api-core==2.30.0
288 # via
289 # google-cloud-core
290 # google-cloud-storage
291 # opencensus
292 google-auth==2.49.1
293 # via
294 # google-api-core
295 # google-cloud-core
296 # google-cloud-storage
297 # runai-model-streamer-gcs
298 google-cloud-core==2.5.0
299 # via google-cloud-storage
300 google-cloud-storage==3.10.1
301 # via runai-model-streamer-gcs
302 google-crc32c==1.8.0
303 # via
304 # google-cloud-storage
305 # google-resumable-media
306 google-resumable-media==2.8.0
307 # via google-cloud-storage
308 googleapis-common-protos==1.73.0
309 # via
310 # google-api-core
311 # opentelemetry-exporter-otlp-proto-grpc
312 # opentelemetry-exporter-otlp-proto-http
313 gpt-oss==0.0.9
314 # via -r requirements/test/rocm.in
315 graphql-core==3.2.8
316 # via hypothesis-graphql
317 greenlet==3.3.2
318 # via sqlalchemy
319 grpcio==1.78.0
320 # via
321 # -c requirements/rocm.txt
322 # -r requirements/test/rocm.in
323 # grpcio-reflection
324 # opentelemetry-exporter-otlp-proto-grpc
325 # ray
326 grpcio-reflection==1.78.0
327 # via
328 # -c requirements/rocm.txt
329 # -r requirements/test/rocm.in
330 h11==0.16.0
331 # via
332 # httpcore
333 # httpcore2
334 # uvicorn
335 h2==4.4.1
336 # via
337 # -r requirements/test/rocm.in
338 # httpx
339 harfile==0.5.0
340 # via schemathesis
341 hf-xet==1.6.0
342 # via huggingface-hub
343 hiredis==3.3.1
344 # via tensorizer
345 hpack==4.2.0
346 # via
347 # -r requirements/test/rocm.in
348 # h2
349 html2text==2025.4.15
350 # via gpt-oss
351 httpcore==1.0.9
352 # via httpx
353 httpcore2==2.12.0
354 # via httpx2
355 httptools==0.7.1
356 # via uvicorn
357 httpx==0.27.2
358 # via
359 # -r requirements/test/rocm.in
360 # anthropic
361 # cohere
362 # fastapi
363 # fastapi-cloud-cli
364 # huggingface-hub
365 # model-hosting-container-standards
366 # openai
367 # perceptron
368 # schemathesis
369 httpx2==2.12.0
370 # via
371 # -r requirements/test/rocm.in
372 # mcp
373 huggingface-hub==1.31.0
374 # via
375 # -c requirements/common.txt
376 # -r requirements/test/../common.txt
377 # accelerate
378 # datasets
379 # evaluate
380 # open-clip-torch
381 # peft
382 # segmentation-models-pytorch
383 # sentence-transformers
384 # timm
385 # tokenizers
386 # transformers
387 # vocos
388 humanize==4.15.0
389 # via runai-model-streamer
390 hyperframe==6.1.0
391 # via h2
392 hypothesis==6.151.9
393 # via
394 # hypothesis-graphql
395 # hypothesis-jsonschema
396 # schemathesis
397 hypothesis-graphql==0.13.0
398 # via schemathesis
399 hypothesis-jsonschema==0.23.1
400 # via schemathesis
401 idna==3.19
402 # via
403 # anyio
404 # email-validator
405 # httpx
406 # httpx2
407 # requests
408 # yarl
409 ijson==3.5.0
410 # via -r requirements/test/../common.txt
411 imagehash==4.3.2
412 # via -r requirements/test/rocm.in
413 imageio==2.37.3
414 # via scikit-image
415 importlib-metadata==8.7.1
416 # via opentelemetry-api
417 inflect==7.5.0
418 # via datamodel-code-generator
419 iniconfig==2.3.0
420 # via pytest
421 instanttensor==0.1.9
422 # via -r requirements/test/rocm.in
423 interegular==0.3.3
424 # via lm-format-enforcer
425 isodate==0.7.2
426 # via azure-storage-blob
427 isort==8.0.1
428 # via datamodel-code-generator
429 jinja2==3.1.6
430 # via
431 # datamodel-code-generator
432 # fastapi
433 # genai-perf
434 # gpt-oss
435 # lm-eval
436 # torch
437 jiter==0.14.0
438 # via
439 # anthropic
440 # openai
441 jiwer==4.0.0
442 # via -r requirements/test/rocm.in
443 jmespath==1.1.0
444 # via
445 # boto3
446 # botocore
447 # model-hosting-container-standards
448 joblib==1.5.3
449 # via
450 # librosa
451 # nltk
452 # scikit-learn
453 jsonschema==4.26.0
454 # via
455 # -c requirements/common.txt
456 # -r requirements/test/../common.txt
457 # hypothesis-jsonschema
458 # mcp
459 # mistral-common
460 # ray
461 jsonschema-rs==0.46.5
462 # via schemathesis
463 jsonschema-specifications==2025.9.1
464 # via jsonschema
465 junit-xml==1.9
466 # via schemathesis
467 jupyter-client==8.10.0
468 # via gpt-oss
469 jupyter-core==5.9.1
470 # via jupyter-client
471 kaldi-native-fbank==1.22.3
472 # via -r requirements/test/rocm.in
473 kaleido==1.0.0
474 # via genai-perf
475 kiwisolver==1.5.0
476 # via matplotlib
477 lark==1.2.2
478 # via
479 # -c requirements/common.txt
480 # -r requirements/test/../common.txt
481 lazy-loader==0.4
482 # via
483 # librosa
484 # scikit-image
485 libnacl==2.1.0
486 # via tensorizer
487 librosa==0.10.2.post1
488 # via -r requirements/test/rocm.in
489 llguidance==1.7.5
490 # via
491 # -c requirements/common.txt
492 # -r requirements/test/../common.txt
493 llvmlite==0.47.0
494 # via numba
495 lm-eval==0.4.12
496 # via -r requirements/test/rocm.in
497 lm-format-enforcer==0.11.3
498 # via
499 # -c requirements/common.txt
500 # -r requirements/test/../common.txt
501 logistro==2.0.1
502 # via
503 # choreographer
504 # kaleido
505 loguru==0.7.3
506 # via compressed-tensors
507 lxml==6.1.3
508 # via
509 # -r requirements/test/rocm.in
510 # blobfile
511 # gpt-oss
512 # sacrebleu
513 mako==1.4.3
514 # via
515 # -r requirements/test/rocm.in
516 # alembic
517 markdown-it-py==4.0.0
518 # via rich
519 markupsafe==3.0.3
520 # via
521 # jinja2
522 # mako
523 # werkzeug
524 matplotlib==3.10.8
525 # via -r requirements/test/rocm.in
526 mbstrdecoder==1.1.4
527 # via
528 # dataproperty
529 # pytablewriter
530 # typepy
531 mcp==2.1.1
532 # via
533 # -c requirements/common.txt
534 # -r requirements/test/../common.txt
535 mcp-types==2.1.1
536 # via mcp
537 mdurl==0.1.2
538 # via markdown-it-py
539 mistral-common==1.11.6
540 # via
541 # -c requirements/common.txt
542 # -r requirements/test/../common.txt
543 # -r requirements/test/rocm.in
544 ml-dtypes==0.5.4
545 # via tilelang
546 model-hosting-container-standards==0.1.14
547 # via
548 # -c requirements/common.txt
549 # -r requirements/test/../common.txt
550 more-itertools==10.8.0
551 # via
552 # inflect
553 # lm-eval
554 mpmath==1.3.0
555 # via sympy
556 msal==1.39.0
557 # via
558 # azure-identity
559 # msal-extensions
560 msal-extensions==1.3.1
561 # via azure-identity
562 msgpack==1.2.3
563 # via
564 # -r requirements/test/rocm.in
565 # librosa
566 # ray
567 msgspec==0.21.0
568 # via -r requirements/test/../common.txt
569 mteb==2.11.5
570 # via -r requirements/test/rocm.in
571 multidict==6.7.1
572 # via
573 # aiohttp
574 # yarl
575 multiprocess==0.70.16
576 # via
577 # -r requirements/test/rocm.in
578 # datasets
579 # evaluate
580 mypy-extensions==1.1.0
581 # via black
582 narwhals==2.18.0
583 # via plotly
584 networkx==3.6.1
585 # via
586 # scikit-image
587 # torch
588 ninja==1.13.0
589 # via -r requirements/test/../common.txt
590 nltk==3.10.3
591 # via
592 # -r requirements/test/rocm.in
593 # rouge-score
594 num2words==0.5.14
595 # via -r requirements/test/rocm.in
596 numba==0.65.0
597 # via
598 # -c requirements/rocm.txt
599 # -r requirements/test/rocm.in
600 # librosa
601 numpy==2.2.6
602 # via
603 # -r requirements/test/../common.txt
604 # accelerate
605 # albumentations
606 # bitsandbytes
607 # bm25s
608 # contourpy
609 # cupy-cuda12x
610 # datasets
611 # decord
612 # einx
613 # encodec
614 # evaluate
615 # fastparquet
616 # genai-perf
617 # gpt-oss
618 # imagehash
619 # imageio
620 # librosa
621 # lm-eval
622 # matplotlib
623 # mistral-common
624 # ml-dtypes
625 # mteb
626 # numba
627 # opencv-python-headless
628 # optuna
629 # pandas
630 # patsy
631 # peft
632 # perceptron
633 # pytrec-eval-terrier
634 # pywavelets
635 # rouge-score
636 # runai-model-streamer
637 # sacrebleu
638 # scikit-image
639 # scikit-learn
640 # scipy
641 # segmentation-models-pytorch
642 # sentence-transformers
643 # soundfile
644 # soxr
645 # statsmodels
646 # tensorizer
647 # tifffile
648 # tilelang
649 # torchvision
650 # transformers
651 # tritonclient
652 # vocos
653 # xgrammar
654 open-clip-torch==2.32.0
655 # via -r requirements/test/rocm.in
656 openai==2.31.0
657 # via
658 # -c requirements/common.txt
659 # -r requirements/test/../common.txt
660 # gpt-oss
661 opencensus==0.11.4
662 # via ray
663 opencensus-context==0.1.3
664 # via opencensus
665 opencv-python-headless==4.13.0.92
666 # via
667 # -c requirements/common.txt
668 # -r requirements/test/../common.txt
669 # albumentations
670 # mistral-common
671 openpyxl==3.1.5
672 # via -r requirements/test/rocm.in
673 opentelemetry-api==1.40.0
674 # via
675 # -c requirements/common.txt
676 # -r requirements/test/../common.txt
677 # mcp
678 # opentelemetry-exporter-otlp-proto-grpc
679 # opentelemetry-exporter-otlp-proto-http
680 # opentelemetry-exporter-prometheus
681 # opentelemetry-sdk
682 # opentelemetry-semantic-conventions
683 opentelemetry-exporter-otlp==1.40.0
684 # via
685 # -c requirements/common.txt
686 # -r requirements/test/../common.txt
687 opentelemetry-exporter-otlp-proto-common==1.40.0
688 # via
689 # opentelemetry-exporter-otlp-proto-grpc
690 # opentelemetry-exporter-otlp-proto-http
691 opentelemetry-exporter-otlp-proto-grpc==1.40.0
692 # via opentelemetry-exporter-otlp
693 opentelemetry-exporter-otlp-proto-http==1.40.0
694 # via opentelemetry-exporter-otlp
695 opentelemetry-exporter-prometheus==0.61b0
696 # via ray
697 opentelemetry-proto==1.40.0
698 # via
699 # opentelemetry-exporter-otlp-proto-common
700 # opentelemetry-exporter-otlp-proto-grpc
701 # opentelemetry-exporter-otlp-proto-http
702 # ray
703 opentelemetry-sdk==1.40.0
704 # via
705 # -c requirements/common.txt
706 # -r requirements/test/../common.txt
707 # opentelemetry-exporter-otlp-proto-grpc
708 # opentelemetry-exporter-otlp-proto-http
709 # opentelemetry-exporter-prometheus
710 # opentelemetry-semantic-conventions-ai
711 # ray
712 opentelemetry-semantic-conventions==0.61b0
713 # via
714 # opentelemetry-sdk
715 # opentelemetry-semantic-conventions-ai
716 opentelemetry-semantic-conventions-ai==0.5.1
717 # via
718 # -c requirements/common.txt
719 # -r requirements/test/../common.txt
720 optuna==3.6.1
721 # via genai-perf
722 orjson==3.11.7
723 # via
724 # -r requirements/test/rocm.in
725 # genai-perf
726 # kaleido
727 oss-harmony==0.0.11
728 # via
729 # -c requirements/common.txt
730 # -r requirements/test/../common.txt
731 outlines-core==0.2.14
732 # via
733 # -c requirements/common.txt
734 # -r requirements/test/../common.txt
735 packaging==26.2
736 # via
737 # -c requirements/rocm.txt
738 # accelerate
739 # bitsandbytes
740 # black
741 # datasets
742 # evaluate
743 # fastparquet
744 # huggingface-hub
745 # kaleido
746 # lazy-loader
747 # lm-format-enforcer
748 # matplotlib
749 # optuna
750 # peft
751 # plotly
752 # pooch
753 # pytest
754 # pytest-rerunfailures
755 # ray
756 # scikit-image
757 # statsmodels
758 # transformers
759 # typepy
760 pandas==3.0.1
761 # via
762 # datasets
763 # evaluate
764 # fastparquet
765 # genai-perf
766 # gpt-oss
767 # statsmodels
768 partial-json-parser==0.2.1.1.post7
769 # via -r requirements/test/../common.txt
770 pathspec==1.0.4
771 # via black
772 pathvalidate==3.3.1
773 # via pytablewriter
774 patsy==1.0.2
775 # via statsmodels
776 peft==0.19.1
777 # via -r requirements/test/rocm.in
778 perceptron==0.1.4
779 # via -r requirements/test/rocm.in
780 perf-analyzer==0.1.0
781 # via genai-perf
782 pillow==12.3.0
783 # via
784 # -c requirements/common.txt
785 # -r requirements/test/../common.txt
786 # genai-perf
787 # imagehash
788 # imageio
789 # matplotlib
790 # mistral-common
791 # perceptron
792 # scikit-image
793 # segmentation-models-pytorch
794 # torchvision
795 platformdirs==4.3.6
796 # via
797 # black
798 # jupyter-core
799 # pooch
800 # python-discovery
801 # virtualenv
802 plotly==6.6.0
803 # via
804 # -r requirements/test/rocm.in
805 # genai-perf
806 pluggy==1.6.0
807 # via
808 # pytest
809 # pytest-cov
810 polars==1.39.3
811 # via mteb
812 polars-runtime-32==1.39.3
813 # via polars
814 pooch==1.8.2
815 # via librosa
816 portalocker==3.2.0
817 # via sacrebleu
818 pqdm==0.2.0
819 # via -r requirements/test/rocm.in
820 prometheus-client==0.24.1
821 # via
822 # -c requirements/common.txt
823 # -r requirements/test/../common.txt
824 # opentelemetry-exporter-prometheus
825 # prometheus-fastapi-instrumentator
826 # ray
827 prometheus-fastapi-instrumentator==8.0.0
828 # via
829 # -c requirements/common.txt
830 # -r requirements/test/../common.txt
831 propcache==0.4.1
832 # via
833 # aiohttp
834 # yarl
835 proto-plus==1.27.1
836 # via google-api-core
837 protobuf==6.33.6
838 # via
839 # -c requirements/common.txt
840 # -r requirements/test/../common.txt
841 # google-api-core
842 # googleapis-common-protos
843 # grpcio-reflection
844 # opentelemetry-proto
845 # proto-plus
846 # ray
847 # tensorizer
848 psutil==7.2.2
849 # via
850 # -r requirements/test/../common.txt
851 # accelerate
852 # peft
853 # tensorizer
854 # tilelang
855 py==1.11.0
856 # via pytest-forked
857 py-cpuinfo==9.0.0
858 # via -r requirements/test/../common.txt
859 py-spy==0.4.1
860 # via ray
861 pyarrow==23.0.1
862 # via
863 # -r requirements/test/rocm.in
864 # datasets
865 # genai-perf
866 pyasn1==0.6.4
867 # via
868 # -r requirements/test/rocm.in
869 # pyasn1-modules
870 pyasn1-modules==0.4.2
871 # via google-auth
872 pybase64==1.4.3
873 # via -r requirements/test/../common.txt
874 pycountry==26.2.16
875 # via pydantic-extra-types
876 pycparser==3.0
877 # via cffi
878 pycryptodomex==3.23.0
879 # via blobfile
880 pydantic==2.12.5
881 # via
882 # -c requirements/common.txt
883 # -r requirements/test/../common.txt
884 # albumentations
885 # anthropic
886 # cohere
887 # compressed-tensors
888 # datamodel-code-generator
889 # fastapi
890 # fastapi-cloud-cli
891 # gpt-oss
892 # lm-format-enforcer
893 # mcp
894 # mcp-types
895 # mistral-common
896 # model-hosting-container-standards
897 # mteb
898 # openai
899 # oss-harmony
900 # pydantic-extra-types
901 # pydantic-settings
902 # ray
903 # xgrammar
904 pydantic-core==2.41.5
905 # via
906 # cohere
907 # pydantic
908 pydantic-extra-types==2.11.1
909 # via
910 # fastapi
911 # mistral-common
912 pydantic-settings==2.15.0
913 # via
914 # -r requirements/test/rocm.in
915 # fastapi
916 pygments==2.21.0
917 # via
918 # -r requirements/test/rocm.in
919 # pytest
920 # rich
921 pyjwt==2.15.1
922 # via
923 # -r requirements/test/rocm.in
924 # mcp
925 # msal
926 pyparsing==3.3.2
927 # via matplotlib
928 pyrate-limiter==4.4.0
929 # via schemathesis
930 pystemmer==3.0.0
931 # via mteb
932 pytablewriter==1.2.1
933 # via lm-eval
934 pytest==9.1.0
935 # via
936 # -r requirements/test/rocm.in
937 # buildkite-test-collector
938 # genai-perf
939 # pytest-asyncio
940 # pytest-cov
941 # pytest-forked
942 # pytest-mock
943 # pytest-rerunfailures
944 # pytest-shard
945 # pytest-timeout
946 # schemathesis
947 pytest-asyncio==1.4.0
948 # via -r requirements/test/rocm.in
949 pytest-cov==6.3.0
950 # via -r requirements/test/rocm.in
951 pytest-forked==1.6.0
952 # via -r requirements/test/rocm.in
953 pytest-mock==3.15.1
954 # via genai-perf
955 pytest-rerunfailures==14.0
956 # via -r requirements/test/rocm.in
957 pytest-shard==0.1.2
958 # via -r requirements/test/rocm.in
959 pytest-timeout==2.3.1
960 # via -r requirements/test/rocm.in
961 python-dateutil==2.9.0.post0
962 # via
963 # botocore
964 # jupyter-client
965 # matplotlib
966 # pandas
967 # typepy
968 python-discovery==1.2.0
969 # via virtualenv
970 python-dotenv==1.2.2
971 # via
972 # pydantic-settings
973 # uvicorn
974 python-json-logger==4.1.0
975 # via -r requirements/test/../common.txt
976 python-multipart==0.0.32
977 # via
978 # -r requirements/test/rocm.in
979 # fastapi
980 # mcp
981 python-rapidjson==1.23
982 # via tritonclient
983 pytokens==0.4.1
984 # via black
985 pytrec-eval-terrier==0.5.10
986 # via mteb
987 pytz==2026.1.post1
988 # via typepy
989 pywavelets==1.9.0
990 # via imagehash
991 pyyaml==6.0.3
992 # via
993 # -r requirements/test/../common.txt
994 # accelerate
995 # albumentations
996 # datamodel-code-generator
997 # datasets
998 # genai-perf
999 # huggingface-hub
1000 # lm-format-enforcer
1001 # optuna
1002 # peft
1003 # ray
1004 # responses
1005 # schemathesis
1006 # timm
1007 # transformers
1008 # uvicorn
1009 # vocos
1010 pyzmq==27.1.0
1011 # via
1012 # -c requirements/common.txt
1013 # -r requirements/test/../common.txt
1014 # jupyter-client
1015 rapidfuzz==3.12.1
1016 # via
1017 # -r requirements/test/rocm.in
1018 # jiwer
1019 ray==2.55.1
1020 # via -r requirements/test/rocm.in
1021 redis==7.3.0
1022 # via tensorizer
1023 referencing==0.37.0
1024 # via
1025 # jsonschema
1026 # jsonschema-specifications
1027 regex==2026.2.28
1028 # via
1029 # -r requirements/test/../common.txt
1030 # nltk
1031 # open-clip-torch
1032 # sacrebleu
1033 # tiktoken
1034 # transformers
1035 requests==2.34.2
1036 # via
1037 # -c requirements/common.txt
1038 # -r requirements/test/../common.txt
1039 # azure-core
1040 # buildkite-test-collector
1041 # cohere
1042 # datasets
1043 # docker
1044 # evaluate
1045 # google-api-core
1046 # google-cloud-storage
1047 # gpt-oss
1048 # lm-eval
1049 # mistral-common
1050 # msal
1051 # mteb
1052 # opentelemetry-exporter-otlp-proto-http
1053 # pooch
1054 # ray
1055 # responses
1056 # schemathesis
1057 # starlette-testclient
1058 # tiktoken
1059 responses==0.26.0
1060 # via genai-perf
1061 rich==14.3.3
1062 # via
1063 # genai-perf
1064 # mteb
1065 # perceptron
1066 # rich-toolkit
1067 # schemathesis
1068 # typer
1069 rich-toolkit==0.19.7
1070 # via
1071 # fastapi-cli
1072 # fastapi-cloud-cli
1073 rignore==0.7.6
1074 # via fastapi-cloud-cli
1075 rouge-score==0.1.2
1076 # via lm-eval
1077 rpds-py==0.30.0
1078 # via
1079 # jsonschema
1080 # referencing
1081 runai-model-streamer==0.15.7
1082 # via
1083 # -c requirements/rocm.txt
1084 # -r requirements/test/rocm.in
1085 runai-model-streamer-azure==0.15.7
1086 # via runai-model-streamer
1087 runai-model-streamer-gcs==0.15.7
1088 # via runai-model-streamer
1089 runai-model-streamer-s3==0.15.7
1090 # via runai-model-streamer
1091 s3transfer==0.16.0
1092 # via boto3
1093 sacrebleu==2.6.0
1094 # via lm-eval
1095 safetensors==0.8.0
1096 # via
1097 # -c requirements/common.txt
1098 # -r requirements/test/../common.txt
1099 # accelerate
1100 # open-clip-torch
1101 # peft
1102 # segmentation-models-pytorch
1103 # timm
1104 # transformers
1105 schemathesis==4.21.6
1106 # via -r requirements/test/rocm.in
1107 scikit-image==0.26.0
1108 # via albumentations
1109 scikit-learn==1.8.0
1110 # via
1111 # albumentations
1112 # librosa
1113 # lm-eval
1114 # mteb
1115 # sentence-transformers
1116 scipy==1.17.1
1117 # via
1118 # albumentations
1119 # bm25s
1120 # imagehash
1121 # librosa
1122 # mteb
1123 # pytrec-eval-terrier
1124 # scikit-image
1125 # scikit-learn
1126 # sentence-transformers
1127 # statsmodels
1128 # vocos
1129 segmentation-models-pytorch==0.5.0
1130 # via -r requirements/test/rocm.in
1131 sentence-transformers==5.3.0
1132 # via
1133 # -r requirements/test/rocm.in
1134 # mteb
1135 sentencepiece==0.2.1
1136 # via -r requirements/test/../common.txt
1137 sentry-sdk==2.55.0
1138 # via fastapi-cloud-cli
1139 setproctitle==1.3.7
1140 # via -r requirements/test/../common.txt
1141 setuptools==79.0.1
1142 # via
1143 # -c requirements/common.txt
1144 # -c requirements/rocm.txt
1145 # -r requirements/test/../common.txt
1146 # model-hosting-container-standards
1147 # pytablewriter
1148 # torch
1149 shellingham==1.5.4
1150 # via
1151 # perceptron
1152 # typer
1153 simplejson==3.20.2
1154 # via choreographer
1155 six==1.17.0
1156 # via
1157 # -c requirements/common.txt
1158 # -r requirements/test/../common.txt
1159 # junit-xml
1160 # opencensus
1161 # python-dateutil
1162 # rouge-score
1163 smart-open==7.5.1
1164 # via ray
1165 sniffio==1.3.1
1166 # via
1167 # anthropic
1168 # httpx
1169 # openai
1170 sortedcontainers==2.4.0
1171 # via hypothesis
1172 soundfile==0.13.1
1173 # via
1174 # -r requirements/test/rocm.in
1175 # genai-perf
1176 # librosa
1177 # mistral-common
1178 soxr==0.5.0.post1
1179 # via
1180 # librosa
1181 # mistral-common
1182 sqlalchemy==2.0.48
1183 # via
1184 # alembic
1185 # optuna
1186 sqlitedict==2.1.0
1187 # via lm-eval
1188 sse-starlette==3.3.4
1189 # via mcp
1190 starlette==1.3.1
1191 # via
1192 # -c requirements/common.txt
1193 # -r requirements/test/../common.txt
1194 # fastapi
1195 # mcp
1196 # model-hosting-container-standards
1197 # prometheus-fastapi-instrumentator
1198 # sse-starlette
1199 # starlette-testclient
1200 starlette-testclient==0.4.1
1201 # via schemathesis
1202 statsmodels==0.14.6
1203 # via genai-perf
1204 structlog==26.1.0
1205 # via gpt-oss
1206 supervisor==4.3.0
1207 # via model-hosting-container-standards
1208 sympy==1.14.0
1209 # via
1210 # einx
1211 # torch
1212 tabledata==1.3.4
1213 # via pytablewriter
1214 tabulate==0.10.0
1215 # via sacrebleu
1216 tblib==3.1.0
1217 # via -r requirements/test/rocm.in
1218 tcolorpy==0.1.7
1219 # via pytablewriter
1220 tenacity==9.1.4
1221 # via
1222 # gpt-oss
1223 # lm-eval
1224 # schemathesis
1225 tensorizer==2.10.1
1226 # via
1227 # -c requirements/rocm.txt
1228 # -r requirements/test/rocm.in
1229 termcolor==3.3.0
1230 # via gpt-oss
1231 threadpoolctl==3.6.0
1232 # via scikit-learn
1233 tifffile==2026.3.3
1234 # via scikit-image
1235 tiktoken==0.12.0
1236 # via
1237 # -c requirements/common.txt
1238 # -r requirements/test/../common.txt
1239 # gpt-oss
1240 # lm-eval
1241 # mistral-common
1242 tilelang==0.1.10
1243 # via
1244 # -c requirements/rocm.txt
1245 # -r requirements/test/rocm.in
1246 timm==1.0.17
1247 # via
1248 # -c requirements/rocm.txt
1249 # -r requirements/test/rocm.in
1250 # open-clip-torch
1251 # segmentation-models-pytorch
1252 tokenizers==0.23.1
1253 # via
1254 # -c requirements/common.txt
1255 # -r requirements/test/../common.txt
1256 # cohere
1257 # transformers
1258 torch-c-dlpack-ext==0.1.5
1259 # via tilelang
1260 tornado==6.5.10
1261 # via jupyter-client
1262 tqdm==4.67.3
1263 # via
1264 # -r requirements/test/../common.txt
1265 # datasets
1266 # evaluate
1267 # gpt-oss
1268 # huggingface-hub
1269 # lm-eval
1270 # mteb
1271 # nltk
1272 # open-clip-torch
1273 # openai
1274 # optuna
1275 # peft
1276 # pqdm
1277 # segmentation-models-pytorch
1278 # sentence-transformers
1279 # tilelang
1280 # transformers
1281 traitlets==5.16.1
1282 # via
1283 # jupyter-client
1284 # jupyter-core
1285 transformers==5.17.0
1286 # via
1287 # -c requirements/common.txt
1288 # -r requirements/test/../common.txt
1289 # -r requirements/test/rocm.in
1290 # compressed-tensors
1291 # genai-perf
1292 # peft
1293 # sentence-transformers
1294 # transformers-stream-generator
1295 # xgrammar
1296 transformers-stream-generator==0.0.5
1297 # via -r requirements/test/rocm.in
1298 tritonclient==2.66.0
1299 # via -r requirements/test/rocm.in
1300 truststore==0.10.4
1301 # via
1302 # httpcore2
1303 # httpx2
1304 typeguard==4.5.1
1305 # via inflect
1306 typepy==1.3.4
1307 # via
1308 # dataproperty
1309 # pytablewriter
1310 # tabledata
1311 typer==0.24.1
1312 # via
1313 # fastapi-cli
1314 # fastapi-cloud-cli
1315 # fastsafetensors
1316 # perceptron
1317 # transformers
1318 types-requests==2.33.0.20260712
1319 # via cohere
1320 typing-extensions==4.15.0
1321 # via
1322 # -c requirements/common.txt
1323 # -r requirements/test/../common.txt
1324 # aiohttp
1325 # aiosignal
1326 # albumentations
1327 # alembic
1328 # anthropic
1329 # anyio
1330 # apache-tvm-ffi
1331 # azure-core
1332 # azure-identity
1333 # azure-storage-blob
1334 # chz
1335 # cohere
1336 # fastapi
1337 # grpcio
1338 # httpx2
1339 # huggingface-hub
1340 # jupyter-client
1341 # librosa
1342 # lm-eval
1343 # mcp
1344 # mcp-types
1345 # mistral-common
1346 # mteb
1347 # openai
1348 # opentelemetry-api
1349 # opentelemetry-exporter-otlp-proto-grpc
1350 # opentelemetry-exporter-otlp-proto-http
1351 # opentelemetry-sdk
1352 # opentelemetry-semantic-conventions
1353 # pqdm
1354 # pydantic
1355 # pydantic-core
1356 # pydantic-extra-types
1357 # pytest-asyncio
1358 # referencing
1359 # rich-toolkit
1360 # schemathesis
1361 # sentence-transformers
1362 # sqlalchemy
1363 # starlette
1364 # tilelang
1365 # torch
1366 # typeguard
1367 # typing-inspection
1368 # xgrammar
1369 typing-inspection==0.4.2
1370 # via
1371 # fastapi
1372 # mcp
1373 # pydantic
1374 # pydantic-settings
1375 urllib3==2.8.0
1376 # via
1377 # -r requirements/test/rocm.in
1378 # blobfile
1379 # botocore
1380 # docker
1381 # requests
1382 # responses
1383 # sentry-sdk
1384 # tritonclient
1385 # types-requests
1386 uvicorn==0.42.0
1387 # via
1388 # fastapi
1389 # fastapi-cli
1390 # fastapi-cloud-cli
1391 # gpt-oss
1392 # mcp
1393 uvloop==0.22.1
1394 # via uvicorn
1395 vector-quantize-pytorch==1.28.0
1396 # via -r requirements/test/rocm.in
1397 virtualenv==21.2.0
1398 # via
1399 # -r requirements/test/rocm.in
1400 # ray
1401 vocos==0.1.0
1402 # via -r requirements/test/rocm.in
1403 watchfiles==1.1.1
1404 # via
1405 # -r requirements/test/../common.txt
1406 # uvicorn
1407 wcwidth==0.6.0
1408 # via ftfy
1409 websockets==16.0
1410 # via uvicorn
1411 werkzeug==3.1.6
1412 # via
1413 # -r requirements/test/rocm.in
1414 # schemathesis
1415 word2number==1.1
1416 # via lm-eval
1417 wrapt==2.1.2
1418 # via smart-open
1419 xgrammar==0.2.8
1420 # via
1421 # -c requirements/common.txt
1422 # -r requirements/test/../common.txt
1423 xxhash==3.6.0
1424 # via
1425 # datasets
1426 # evaluate
1427 yarl==1.23.0
1428 # via aiohttp
1429 z3-solver==4.15.4.0
1430 # via tilelang
1431 zipp==3.23.0
1432 # via importlib-metadata
1433
1434 # The following packages were excluded from the output:
1435 # torch
1436 # torchvision
1437 # torchaudio
1438 # triton
1439 # cuda-bindings
1440 # cuda-pathfinder
1441 # cuda-toolkit
1442 # cupy-cuda12x
1443 # nvidia-cublas
1444 # nvidia-cuda-cupti
1445 # nvidia-cuda-nvrtc
1446 # nvidia-cuda-runtime
1447 # nvidia-cufft
1448 # nvidia-cufile
1449 # nvidia-curand
1450 # nvidia-cusolver
1451 # nvidia-cusparse
1452 # nvidia-nvjitlink
1453 # nvidia-nvtx
1454 # nvidia-cudnn-cu13
1455 # nvidia-cusparselt-cu13
1456 # nvidia-nccl-cu13
1457 # nvidia-nvshmem-cu13