maxtext

Monthly downloads

1.4K

Weekly downloads

309

GitHub stars

2.4K

Downloads in the 30 and 7 days to October 10, 2026. Refreshed weekly.

Package Information

Latest Version

0.2.5

Requires Python

>=3.12

Licence

Apache-2.0

Last Release

2 weeks ago

First Release

September 15, 2025

Total Releases

Python Versions

3.12

No reported malware or high/critical advisories · Details

The current version has no OSV.dev malware reports and no critical or high-severity vulnerabilities.

Package Insights

What's this?

Downloads by OS

Linux 14

Python Versions

3.12

92.9% 13

3.13

7.1% 1

Top Countries

Weekly Downloads

Last 7 days

309

Last 30 days

1,385

Peak week (to 2026-10-03)

559

2026-09-25 – 2026-10-10. Enable JavaScript for the interactive chart.

GitHub Stars

Current

2,441

Earliest (2026-09-26)

2,430

Change

+11

2026-09-26 – 2026-10-11. Enable JavaScript for the interactive chart.

Dependencies

1000 optional dependencies

absl-py>=2.5.0[cuda12]
extra == "cuda12"
absl-py>=2.3.1[runner]
extra == "runner"
absl-py>=2.5.0[tpu]
extra == "tpu"
absl-py>=2.5.0[tpu-post-train]
extra == "tpu-post-train"
accelerate>=1.15.0[tpu-post-train]
extra == "tpu-post-train"
aiofiles>=25.1.0[cuda12]
extra == "cuda12"
aiofiles>=25.1.0[tpu]
extra == "tpu"
aiofiles>=25.1.0[tpu-post-train]
extra == "tpu-post-train"
aiohappyeyeballs>=2.7.1[cuda12]
extra == "cuda12"
aiohappyeyeballs>=2.7.1[tpu]
extra == "tpu"
aiohappyeyeballs>=2.7.1[tpu-post-train]
extra == "tpu-post-train"
aiohttp>=3.14.3[cuda12]
extra == "cuda12"
aiohttp>=3.14.3[tpu]
extra == "tpu"
aiohttp>=3.14.3[tpu-post-train]
extra == "tpu-post-train"
aiohttp-cors>=0.1.0[tpu-post-train]
extra == "tpu-post-train"
aiosignal>=1.4.0[cuda12]
extra == "cuda12"
aiosignal>=1.4.0[tpu]
extra == "tpu"
aiosignal>=1.4.0[tpu-post-train]
extra == "tpu-post-train"
annotated-doc>=0.0.5[cuda12]
extra == "cuda12"
annotated-doc>=0.0.5[tpu]
extra == "tpu"
annotated-doc>=0.0.5[tpu-post-train]
extra == "tpu-post-train"
annotated-types>=0.8.0[cuda12]
extra == "cuda12"
annotated-types>=0.7.0[runner]
extra == "runner"
annotated-types>=0.8.0[tpu]
extra == "tpu"
annotated-types>=0.8.0[tpu-post-train]
extra == "tpu-post-train"
anthropic>=1.7.0[tpu-post-train]
extra == "tpu-post-train"
antlr4-python3-runtime>=4.9.3[cuda12]
extra == "cuda12"
antlr4-python3-runtime>=4.9.3[runner]
extra == "runner"
antlr4-python3-runtime>=4.9.3[tpu]
extra == "tpu"
antlr4-python3-runtime>=4.9.3[tpu-post-train]
extra == "tpu-post-train"
anyio>=4.15.1[cuda12]
extra == "cuda12"
anyio>=4.12.1[runner]
extra == "runner"
anyio>=4.15.1[tpu]
extra == "tpu"
anyio>=4.15.1[tpu-post-train]
extra == "tpu-post-train"
apache-tvm-ffi>=0.1.14.post0[cuda12]
extra == "cuda12"
apache-tvm-ffi>=0.1.14.post0[tpu-post-train]
extra == "tpu-post-train"
appnope>=1.0.0[tpu-post-train]
sys_platform == "darwin" and extra == "tpu-post-train"
aqtp>=0.9.0[cuda12]
extra == "cuda12"
aqtp>=0.9.0[tpu]
extra == "tpu"
aqtp>=0.9.0[tpu-post-train]
extra == "tpu-post-train"
argcomplete>=3.6.3[runner]
extra == "runner"
array-record>=0.8.3[cuda12]
extra == "cuda12"
array-record>=0.8.3[tpu]
extra == "tpu"
array-record>=0.8.3[tpu-post-train]
extra == "tpu-post-train"
astor>=0.8.1[tpu-post-train]
extra == "tpu-post-train"
astroid>=4.0.4[cuda12]
extra == "cuda12"
astroid>=4.0.4[tpu]
extra == "tpu"
astroid>=4.0.4[tpu-post-train]
extra == "tpu-post-train"
asttokens>=3.0.2[tpu-post-train]
extra == "tpu-post-train"
astunparse>=1.6.3[cuda12]
extra == "cuda12"
astunparse>=1.6.3[runner]
extra == "runner"
astunparse>=1.6.3[tpu]
extra == "tpu"
astunparse>=1.6.3[tpu-post-train]
extra == "tpu-post-train"
attrs>=26.1.0[cuda12]
extra == "cuda12"
attrs>=25.4.0[runner]
extra == "runner"
attrs>=26.1.0[tpu]
extra == "tpu"
attrs>=26.1.0[tpu-post-train]
extra == "tpu-post-train"
auditwheel>=6.8.1[cuda12]
extra == "cuda12"
auditwheel>=6.5.0[runner]
extra == "runner"
auditwheel>=6.8.1[tpu]
extra == "tpu"
auditwheel>=6.7.0[tpu-post-train]
extra == "tpu-post-train"
black>=26.5.1[cuda12]
extra == "cuda12"
black>=26.5.1[tpu]
extra == "tpu"
black>=26.5.1[tpu-post-train]
extra == "tpu-post-train"
blake3>=1.0.9[tpu-post-train]
extra == "tpu-post-train"
blobfile>=3.3.0[cuda12]
extra == "cuda12"
blobfile>=3.3.0[tpu]
extra == "tpu"
blobfile>=3.3.0[tpu-post-train]
extra == "tpu-post-train"
boto>=2.35.0[tpu-post-train]
extra == "tpu-post-train"
boto3>=1.34.0[tpu-post-train]
extra == "tpu-post-train"
botocore>=1.34.63[tpu-post-train]
extra == "tpu-post-train"
build>=1.5.0[cuda12]
extra == "cuda12"
build>=1.3.0[runner]
extra == "runner"
build>=1.5.0[tpu]
extra == "tpu"
build>=1.5.0[tpu-post-train]
extra == "tpu-post-train"
cachetools>=7.2.0[tpu-post-train]
extra == "tpu-post-train"
cbor2>=6.1.4[tpu-post-train]
extra == "tpu-post-train"
certifi>=2026.7.22[cuda12]
extra == "cuda12"
certifi>=2025.11.12[runner]
extra == "runner"
certifi>=2026.7.22[tpu]
extra == "tpu"
certifi>=2026.6.17[tpu-post-train]
extra == "tpu-post-train"
cffi>=2.1.1[cuda12]
platform_python_implementation != "PyPy" and extra == "cuda12"
cffi>=2.0.0[runner]
platform_python_implementation != "PyPy" and extra == "runner"
cffi>=2.1.1[tpu]
platform_python_implementation != "PyPy" and extra == "tpu"
cffi>=2.1.1[tpu-post-train]
(implementation_name == "pypy" or platform_python_implementation != "PyPy") and extra == "tpu-post-train"
cfgv>=3.5.0[cuda12]
extra == "cuda12"
cfgv>=3.5.0[tpu]
extra == "tpu"
cfgv>=3.5.0[tpu-post-train]
extra == "tpu-post-train"
charset-normalizer>=3.5.1[cuda12]
extra == "cuda12"
charset-normalizer>=3.4.4[runner]
extra == "runner"
charset-normalizer>=3.5.1[tpu]
extra == "tpu"
charset-normalizer>=3.4.9[tpu-post-train]
extra == "tpu-post-train"
cheroot>=11.1.2[cuda12]
extra == "cuda12"
cheroot>=11.1.2[tpu]
extra == "tpu"
cheroot>=11.1.2[tpu-post-train]
extra == "tpu-post-train"
chex>=0.1.92[cuda12]
extra == "cuda12"
chex>=0.1.92[tpu]
extra == "tpu"
chex>=0.1.92[tpu-post-train]
extra == "tpu-post-train"
click>=8.5.0[cuda12]
extra == "cuda12"
click>=8.5.0[tpu]
extra == "tpu"
click>=8.5.0[tpu-post-train]
extra == "tpu-post-train"
cloud-accelerator-diagnostics>=0.1.1[cuda12]
extra == "cuda12"
cloud-accelerator-diagnostics>=0.1.1[runner]
extra == "runner"
cloud-accelerator-diagnostics>=0.1.1[tpu]
extra == "tpu"
cloud-accelerator-diagnostics>=0.1.1[tpu-post-train]
extra == "tpu-post-train"
cloudpickle>=3.1.2[cuda12]
extra == "cuda12"
cloudpickle>=3.1.2[runner]
extra == "runner"
cloudpickle>=3.1.2[tpu]
extra == "tpu"
cloudpickle>=3.1.2[tpu-post-train]
extra == "tpu-post-train"
clu>=0.0.12[tpu-post-train]
extra == "tpu-post-train"
cmake>=3.26.1[tpu-post-train]
extra == "tpu-post-train"
colorama>=0.4.6[cuda12]
extra == "cuda12"
colorama>=0.4.6[runner]
extra == "runner"
colorama>=0.4.6[tpu]
extra == "tpu"
colorama>=0.4.6[tpu-post-train]
extra == "tpu-post-train"
colorful>=0.0.0[tpu-post-train]
extra == "tpu-post-train"
comm>=0.2.3[tpu-post-train]
extra == "tpu-post-train"
compressed-tensors>=0.17.0[tpu-post-train]
extra == "tpu-post-train"
contourpy>=1.3.3[cuda12]
extra == "cuda12"
contourpy>=1.3.3[runner]
extra == "runner"
contourpy>=1.3.3[tpu]
extra == "tpu"
contourpy>=1.3.3[tpu-post-train]
extra == "tpu-post-train"
coverage>=7.16.1[cuda12]
extra == "cuda12"
coverage>=7.16.1[tpu]
extra == "tpu"
coverage>=7.16.1[tpu-post-train]
extra == "tpu-post-train"
cryptography>=50.0.1[cuda12]
extra == "cuda12"
cryptography>=46.0.5[runner]
extra == "runner"
cryptography>=50.0.1[tpu]
extra == "tpu"
cryptography>=50.0.1[tpu-post-train]
extra == "tpu-post-train"
cuda-bindings>=13.4.2[cuda12]
extra == "cuda12"
cuda-pathfinder>=1.8.2[cuda12]
extra == "cuda12"
cycler>=0.12.1[cuda12]
extra == "cuda12"
cycler>=0.12.1[runner]
extra == "runner"
cycler>=0.12.1[tpu]
extra == "tpu"
cycler>=0.12.1[tpu-post-train]
extra == "tpu-post-train"
datasets>=4.8.5[cuda12]
extra == "cuda12"
datasets>=4.8.5[tpu]
extra == "tpu"
datasets>=5.0.1[tpu-post-train]
extra == "tpu-post-train"
debugpy>=1.8.22[tpu-post-train]
extra == "tpu-post-train"
decorator>=5.3.1[cuda12]
extra == "cuda12"
decorator>=5.3.1[tpu]
extra == "tpu"
decorator>=5.3.1[tpu-post-train]
extra == "tpu-post-train"
defusedxml>=0.7.1[cuda12]
extra == "cuda12"
defusedxml>=0.7.1[tpu]
extra == "tpu"
defusedxml>=0.7.1[tpu-post-train]
extra == "tpu-post-train"
depyf>=0.20.0[tpu-post-train]
extra == "tpu-post-train"
dill>=0.4.1[cuda12]
extra == "cuda12"
dill>=0.4.1[tpu]
extra == "tpu"
dill>=0.4.1[tpu-post-train]
extra == "tpu-post-train"
distlib>=0.4.3[cuda12]
extra == "cuda12"
distlib>=0.4.3[tpu]
extra == "tpu"
distlib>=0.4.3[tpu-post-train]
extra == "tpu-post-train"
distro>=1.9.0[cuda12]
extra == "cuda12"
distro>=1.9.0[runner]
extra == "runner"
distro>=1.9.0[tpu]
extra == "tpu"
distro>=1.9.0[tpu-post-train]
extra == "tpu-post-train"
dm-tree>=0.1.10[tpu-post-train]
extra == "tpu-post-train"
docker>=7.1.0[runner]
extra == "runner"
docstring-parser>=0.18.0[cuda12]
extra == "cuda12"
docstring-parser>=0.17.0[runner]
extra == "runner"
docstring-parser>=0.18.0[tpu]
extra == "tpu"
docstring-parser>=0.18.0[tpu-post-train]
extra == "tpu-post-train"
drjax>=0.2.1[cuda12]
extra == "cuda12"
drjax>=0.2.1[tpu]
extra == "tpu"
drjax>=0.2.1[tpu-post-train]
extra == "tpu-post-train"
durationpy>=0.10[runner]
extra == "runner"
einops>=0.8.2[tpu-post-train]
extra == "tpu-post-train"
einshape>=1.0[cuda12]
extra == "cuda12"
einshape>=1.0[tpu]
extra == "tpu"
einshape>=1.0[tpu-post-train]
extra == "tpu-post-train"
entrypoints>=0.4[tpu-post-train]
extra == "tpu-post-train"
etils>=1.14.0[cuda12]
extra == "cuda12"
etils>=1.13.0[runner]
extra == "runner"
etils>=1.14.0[tpu]
extra == "tpu"
etils>=1.14.0[tpu-post-train]
extra == "tpu-post-train"
evaluate>=0.2.0[cuda12]
extra == "cuda12"
evaluate>=0.2.0[tpu]
extra == "tpu"
evaluate>=0.4.6[tpu-post-train]
extra == "tpu-post-train"
execnet>=2.1.2[cuda12]
extra == "cuda12"
execnet>=2.1.2[runner]
extra == "runner"
execnet>=2.1.2[tpu]
extra == "tpu"
execnet>=2.1.2[tpu-post-train]
extra == "tpu-post-train"
executing>=2.2.1[tpu-post-train]
extra == "tpu-post-train"
fastapi>=0.141.1[cuda12]
extra == "cuda12"
fastapi>=0.141.1[tpu]
extra == "tpu"
fastapi>=0.141.1[tpu-post-train]
extra == "tpu-post-train"
fastjsonschema>=2.22.2[tpu-post-train]
extra == "tpu-post-train"
filelock>=3.32.3[cuda12]
extra == "cuda12"
filelock>=3.20.0[runner]
extra == "runner"
filelock>=3.32.3[tpu]
extra == "tpu"
filelock>=3.30.2[tpu-post-train]
extra == "tpu-post-train"
fire>=0.7.1[cuda12]
extra == "cuda12"
fire>=0.7.1[tpu]
extra == "tpu"
fire>=0.7.1[tpu-post-train]
extra == "tpu-post-train"
flatbuffers>=25.12.19[cuda12]
extra == "cuda12"
flatbuffers>=25.9.23[runner]
extra == "runner"
flatbuffers>=25.12.19[tpu]
extra == "tpu"
flatbuffers>=25.12.19[tpu-post-train]
extra == "tpu-post-train"
flax>=0.12.9[cuda12]
extra == "cuda12"
flax>=0.12.9[tpu]
extra == "tpu"
flax>=0.12.8[tpu-post-train]
extra == "tpu-post-train"
fonttools>=4.63.0[cuda12]
extra == "cuda12"
fonttools>=4.60.1[runner]
extra == "runner"
fonttools>=4.63.0[tpu]
extra == "tpu"
fonttools>=4.63.0[tpu-post-train]
extra == "tpu-post-train"
frozenlist>=1.8.0[cuda12]
extra == "cuda12"
frozenlist>=1.8.0[tpu]
extra == "tpu"
frozenlist>=1.8.0[tpu-post-train]
extra == "tpu-post-train"
fsspec>=2026.2.0[cuda12]
extra == "cuda12"
fsspec>=2025.10.0[runner]
extra == "runner"
fsspec>=2026.2.0[tpu]
extra == "tpu"
fsspec>=2026.2.0[tpu-post-train]
extra == "tpu-post-train"
gast>=0.7.0[cuda12]
extra == "cuda12"
gast>=0.6.0[runner]
extra == "runner"
gast>=0.7.0[tpu]
extra == "tpu"
gast>=0.7.0[tpu-post-train]
extra == "tpu-post-train"
gcsfs>=2026.2.0[cuda12]
extra == "cuda12"
gcsfs>=2026.2.0[tpu]
extra == "tpu"
gcsfs>=2026.2.0[tpu-post-train]
extra == "tpu-post-train"
gepa>=0.1.4[tpu-post-train]
extra == "tpu-post-train"
gguf>=0.19.0[tpu-post-train]
extra == "tpu-post-train"
google-api-core>=2.38.0[cuda12]
extra == "cuda12"
google-api-core>=2.24.1[runner]
extra == "runner"
google-api-core>=2.38.0[tpu]
extra == "tpu"
google-api-core>=2.38.0[tpu-post-train]
extra == "tpu-post-train"
google-api-python-client>=2.200.0[cuda12]
extra == "cuda12"
google-api-python-client>=2.200.0[tpu]
extra == "tpu"
google-api-python-client>=2.200.0[tpu-post-train]
extra == "tpu-post-train"
google-auth>=2.58.0[cuda12]
extra == "cuda12"
google-auth>=2.49.1[runner]
extra == "runner"
google-auth>=2.58.0[tpu]
extra == "tpu"
google-auth>=2.58.0[tpu-post-train]
extra == "tpu-post-train"
google-auth-httplib2>=0.4.2[cuda12]
extra == "cuda12"
google-auth-httplib2>=0.4.2[tpu]
extra == "tpu"
google-auth-httplib2>=0.4.2[tpu-post-train]
extra == "tpu-post-train"
google-auth-oauthlib>=1.4.1[cuda12]
extra == "cuda12"
google-auth-oauthlib>=1.4.1[tpu]
extra == "tpu"
google-auth-oauthlib>=1.4.1[tpu-post-train]
extra == "tpu-post-train"
google-cloud>=0.34.0[runner]
extra == "runner"
google-cloud-aiplatform>=2.1.3[cuda12]
extra == "cuda12"
google-cloud-aiplatform>=1.142.0[runner]
extra == "runner"
google-cloud-aiplatform>=2.1.3[tpu]
extra == "tpu"
google-cloud-aiplatform>=2.1.3[tpu-post-train]
extra == "tpu-post-train"
google-cloud-appengine-logging>=1.10.0[cuda12]
extra == "cuda12"
google-cloud-appengine-logging>=1.10.0[tpu]
extra == "tpu"
google-cloud-appengine-logging>=1.10.0[tpu-post-train]
extra == "tpu-post-train"
google-cloud-audit-log>=0.6.2[cuda12]
extra == "cuda12"
google-cloud-audit-log>=0.6.2[tpu]
extra == "tpu"
google-cloud-audit-log>=0.6.2[tpu-post-train]
extra == "tpu-post-train"
google-cloud-bigquery>=3.45.2[cuda12]
extra == "cuda12"
google-cloud-bigquery>=3.40.1[runner]
extra == "runner"
google-cloud-bigquery>=3.45.2[tpu]
extra == "tpu"
google-cloud-bigquery>=3.45.2[tpu-post-train]
extra == "tpu-post-train"
google-cloud-common>=1.8.0[runner]
extra == "runner"
google-cloud-core>=2.7.0[cuda12]
extra == "cuda12"
google-cloud-core>=2.5.0[runner]
extra == "runner"
google-cloud-core>=2.7.0[tpu]
extra == "tpu"
google-cloud-core>=2.7.0[tpu-post-train]
extra == "tpu-post-train"
google-cloud-filestore>=1.12.0[runner]
extra == "runner"
google-cloud-logging>=3.16.3[cuda12]
extra == "cuda12"
google-cloud-logging>=3.16.3[tpu]
extra == "tpu"
google-cloud-logging>=3.16.3[tpu-post-train]
extra == "tpu-post-train"
google-cloud-mldiagnostics>=1.0.8[cuda12]
extra == "cuda12"
google-cloud-mldiagnostics>=1.0.8[tpu]
extra == "tpu"
google-cloud-mldiagnostics>=1.0.8[tpu-post-train]
extra == "tpu-post-train"
google-cloud-monitoring>=2.31.0[cuda12]
extra == "cuda12"
google-cloud-monitoring>=2.31.0[tpu]
extra == "tpu"
google-cloud-monitoring>=2.31.0[tpu-post-train]
extra == "tpu-post-train"
google-cloud-resource-manager>=1.18.0[cuda12]
extra == "cuda12"
google-cloud-resource-manager>=1.16.0[runner]
extra == "runner"
google-cloud-resource-manager>=1.18.0[tpu]
extra == "tpu"
google-cloud-resource-manager>=1.18.0[tpu-post-train]
extra == "tpu-post-train"
google-cloud-storage>=3.14.1[cuda12]
extra == "cuda12"
google-cloud-storage>=3.4.1[runner]
extra == "runner"
google-cloud-storage>=3.14.1[tpu]
extra == "tpu"
google-cloud-storage>=3.14.1[tpu-post-train]
extra == "tpu-post-train"
google-cloud-storage-control>=1.15.0[cuda12]
extra == "cuda12"
google-cloud-storage-control>=1.15.0[tpu]
extra == "tpu"
google-cloud-storage-control>=1.15.0[tpu-post-train]
extra == "tpu-post-train"
google-crc32c>=1.8.0[cuda12]
extra == "cuda12"
google-crc32c>=1.8.0[runner]
extra == "runner"
google-crc32c>=1.8.0[tpu]
extra == "tpu"
google-crc32c>=1.8.0[tpu-post-train]
extra == "tpu-post-train"
google-genai>=2.24.0[cuda12]
extra == "cuda12"
google-genai>=1.68.0[runner]
extra == "runner"
google-genai>=2.24.0[tpu]
extra == "tpu"
google-genai>=2.24.0[tpu-post-train]
extra == "tpu-post-train"
google-metrax>=0.2.4[tpu-post-train]
extra == "tpu-post-train"
google-pasta>=0.2.0[cuda12]
extra == "cuda12"
google-pasta>=0.2.0[runner]
extra == "runner"
google-pasta>=0.2.0[tpu]
extra == "tpu"
google-pasta>=0.2.0[tpu-post-train]
extra == "tpu-post-train"
google-resumable-media>=2.10.2[cuda12]
extra == "cuda12"
google-resumable-media>=2.8.0[runner]
extra == "runner"
google-resumable-media>=2.10.2[tpu]
extra == "tpu"
google-resumable-media>=2.10.2[tpu-post-train]
extra == "tpu-post-train"
google-tunix>=0.1.7[tpu-post-train]
extra == "tpu-post-train"
googleapis-common-protos>=1.75.3[cuda12]
extra == "cuda12"
googleapis-common-protos>=1.73.0[runner]
extra == "runner"
googleapis-common-protos>=1.75.3[tpu]
extra == "tpu"
googleapis-common-protos>=1.75.3[tpu-post-train]
extra == "tpu-post-train"
grain>=0.2.18[cuda12]
extra == "cuda12"
grain>=0.2.18[tpu]
extra == "tpu"
grain>=0.2.18[tpu-post-train]
extra == "tpu-post-train"
grpc-google-iam-v1>=0.14.5[cuda12]
extra == "cuda12"
grpc-google-iam-v1>=0.14.3[runner]
extra == "runner"
grpc-google-iam-v1>=0.14.5[tpu]
extra == "tpu"
grpc-google-iam-v1>=0.14.5[tpu-post-train]
extra == "tpu-post-train"
grpcio>=1.83.0[cuda12]
extra == "cuda12"
grpcio>=1.76.0[runner]
extra == "runner"
grpcio>=1.83.0[tpu]
extra == "tpu"
grpcio>=1.82.1[tpu-post-train]
extra == "tpu-post-train"
grpcio-status>=1.83.0[cuda12]
extra == "cuda12"
grpcio-status>=1.71.2[runner]
extra == "runner"
grpcio-status>=1.83.0[tpu]
extra == "tpu"
grpcio-status>=1.81.1[tpu-post-train]
extra == "tpu-post-train"
gspread>=6.2.1[tpu-post-train]
extra == "tpu-post-train"
gviz-api>=1.10.0[cuda12]
extra == "cuda12"
gviz-api>=1.10.0[runner]
extra == "runner"
gviz-api>=1.10.0[tpu]
extra == "tpu"
gviz-api>=1.10.0[tpu-post-train]
extra == "tpu-post-train"
h11>=0.16.0[cuda12]
extra == "cuda12"
h11>=0.16.0[runner]
extra == "runner"
h11>=0.16.0[tpu]
extra == "tpu"
h11>=0.16.0[tpu-post-train]
extra == "tpu-post-train"
h5py>=3.14.0[cuda12]
extra == "cuda12"
h5py>=3.15.1[runner]
extra == "runner"
h5py>=3.14.0[tpu]
extra == "tpu"
h5py>=3.14.0[tpu-post-train]
extra == "tpu-post-train"
hf-transfer>=0.1.9[tpu-post-train]
extra == "tpu-post-train"
hf-xet>=1.6.0[cuda12]
(platform_machine == "AMD64" or platform_machine == "aarch64" or platform_machine == "amd64" or platform_machine == "arm64" or platform_machine == "x86_64") and extra == "cuda12"
hf-xet>=1.6.0[tpu]
(platform_machine == "AMD64" or platform_machine == "aarch64" or platform_machine == "amd64" or platform_machine == "arm64" or platform_machine == "x86_64") and extra == "tpu"
hf-xet>=1.6.0[tpu-post-train]
(platform_machine == "AMD64" or platform_machine == "aarch64" or platform_machine == "amd64" or platform_machine == "arm64" or platform_machine == "x86_64") and extra == "tpu-post-train"
httpcore>=1.0.9[cuda12]
extra == "cuda12"
httpcore>=1.0.9[runner]
extra == "runner"
httpcore>=1.0.9[tpu]
extra == "tpu"
httpcore>=1.0.9[tpu-post-train]
extra == "tpu-post-train"
httpcore2>=2.13.0[tpu-post-train]
sys_platform != "emscripten" and extra == "tpu-post-train"
httplib2>=0.32.0[cuda12]
extra == "cuda12"
httplib2>=0.32.0[tpu]
extra == "tpu"
httplib2>=0.32.0[tpu-post-train]
extra == "tpu-post-train"
httpx>=0.28.1[cuda12]
extra == "cuda12"
httpx>=0.28.1[runner]
extra == "runner"
httpx>=0.28.1[tpu]
extra == "tpu"
httpx>=0.28.1[tpu-post-train]
extra == "tpu-post-train"
httpx-sse>=0.4.0[tpu-post-train]
extra == "tpu-post-train"
httpx2>=2.13.0[tpu-post-train]
extra == "tpu-post-train"
httpx2-jsfetch>=1.0[tpu-post-train]
sys_platform == "emscripten" and extra == "tpu-post-train"
huggingface-hub>=1.32.0[cuda12]
extra == "cuda12"
huggingface-hub>=1.32.0[tpu]
extra == "tpu"
huggingface-hub>=1.32.0[tpu-post-train]
extra == "tpu-post-train"
humanize>=4.16.0[cuda12]
extra == "cuda12"
humanize>=4.16.0[tpu]
extra == "tpu"
humanize>=4.16.0[tpu-post-train]
extra == "tpu-post-train"
hypothesis>=6.165.10[cuda12]
extra == "cuda12"
hypothesis>=6.142.1[runner]
extra == "runner"
hypothesis>=6.165.10[tpu]
extra == "tpu"
hypothesis>=6.151.9[tpu-post-train]
extra == "tpu-post-train"
identify>=2.6.19[cuda12]
extra == "cuda12"
identify>=2.6.19[tpu]
extra == "tpu"
identify>=2.6.19[tpu-post-train]
extra == "tpu-post-train"
idna>=3.18[cuda12]
extra == "cuda12"
idna>=3.11[runner]
extra == "runner"
idna>=3.18[tpu]
extra == "tpu"
idna>=3.18[tpu-post-train]
extra == "tpu-post-train"
ijson>=3.5.1[tpu-post-train]
extra == "tpu-post-train"
immutabledict>=4.3.1[cuda12]
extra == "cuda12"
immutabledict>=4.3.1[tpu]
extra == "tpu"
immutabledict>=4.3.1[tpu-post-train]
extra == "tpu-post-train"
importlab>=0.8.1[cuda12]
extra == "cuda12"
importlab>=0.8.1[tpu]
extra == "tpu"
importlab>=0.8.1[tpu-post-train]
extra == "tpu-post-train"
importlib-metadata>=9.0.1[cuda12]
extra == "cuda12"
importlib-resources>=6.5.2[runner]
extra == "runner"
importlib-resources>=7.1.0[tpu-post-train]
extra == "tpu-post-train"
iniconfig>=2.3.0[cuda12]
extra == "cuda12"
iniconfig>=2.3.0[runner]
extra == "runner"
iniconfig>=2.3.0[tpu]
extra == "tpu"
iniconfig>=2.3.0[tpu-post-train]
extra == "tpu-post-train"
interegular>=0.3.2[tpu-post-train]
extra == "tpu-post-train"
ipykernel>=7.3.0[tpu-post-train]
extra == "tpu-post-train"
ipython>=9.17.1[tpu-post-train]
extra == "tpu-post-train"
ipython-pygments-lexers>=1.1.1[tpu-post-train]
extra == "tpu-post-train"
ipywidgets>=8.1.9[tpu-post-train]
extra == "tpu-post-train"
isort>=9.0.1[cuda12]
extra == "cuda12"
isort>=9.0.1[tpu]
extra == "tpu"
isort>=9.0.1[tpu-post-train]
extra == "tpu-post-train"
jaraco-functools>=4.6.0[cuda12]
extra == "cuda12"
jaraco-functools>=4.6.0[tpu]
extra == "tpu"
jaraco-functools>=4.6.0[tpu-post-train]
extra == "tpu-post-train"
jax>=0.11.1[cuda12]
extra == "cuda12"
jax[docs]
extra == "docs"
jax>=0.11.1[tpu]
extra == "tpu"
jax>=0.11.0[tpu-post-train]
extra == "tpu-post-train"
jax-cuda12-pjrt>=0.11.1[cuda12]
sys_platform == "linux" and extra == "cuda12"
jax-cuda12-plugin>=0.11.1[cuda12]
sys_platform == "linux" and extra == "cuda12"
jaxlib>=0.11.1[cuda12]
extra == "cuda12"
jaxlib>=0.8.1[runner]
extra == "runner"
jaxlib>=0.11.1[tpu]
extra == "tpu"
jaxlib>=0.11.0[tpu-post-train]
extra == "tpu-post-train"
jaxtyping>=0.3.11[cuda12]
extra == "cuda12"
jaxtyping>=0.3.11[tpu]
extra == "tpu"
jaxtyping>=0.3.11[tpu-post-train]
extra == "tpu-post-train"
jedi>=0.20.0[tpu-post-train]
extra == "tpu-post-train"
jinja2>=3.1.6[cuda12]
extra == "cuda12"
jinja2>=3.1.6[runner]
extra == "runner"
jinja2>=3.1.6[tpu]
extra == "tpu"
jinja2>=3.1.6[tpu-post-train]
extra == "tpu-post-train"
jiter>=0.17.0[tpu-post-train]
extra == "tpu-post-train"
jmespath>=1.1.0[tpu-post-train]
extra == "tpu-post-train"
joblib>=1.6.0[cuda12]
extra == "cuda12"
joblib>=1.6.0[tpu]
extra == "tpu"
joblib>=1.6.0[tpu-post-train]
extra == "tpu-post-train"
jsonlines>=4.0.0[cuda12]
extra == "cuda12"
jsonlines>=4.0.0[tpu]
extra == "tpu"
jsonlines>=4.0.0[tpu-post-train]
extra == "tpu-post-train"
jsonschema>=4.26.0[tpu-post-train]
extra == "tpu-post-train"
jsonschema-specifications>=2025.9.1[tpu-post-train]
extra == "tpu-post-train"
jupyter-client>=8.10.0[tpu-post-train]
extra == "tpu-post-train"
jupyter-core>=5.9.1[tpu-post-train]
extra == "tpu-post-train"
jupyterlab-widgets>=3.0.17[tpu-post-train]
extra == "tpu-post-train"
kagglehub>=1.0.2[tpu-post-train]
extra == "tpu-post-train"
kagglesdk>=0.1.37[tpu-post-train]
extra == "tpu-post-train"
keras>=3.15.1[cuda12]
extra == "cuda12"
keras>=3.12.0[runner]
extra == "runner"
keras>=3.15.1[tpu]
extra == "tpu"
keras>=3.15.0[tpu-post-train]
extra == "tpu-post-train"
kiwisolver>=1.5.0[cuda12]
extra == "cuda12"
kiwisolver>=1.4.9[runner]
extra == "runner"
kiwisolver>=1.5.0[tpu]
extra == "tpu"
kiwisolver>=1.5.0[tpu-post-train]
extra == "tpu-post-train"
kubernetes>=31.0.0[runner]
extra == "runner"
lark>=1.2.2[tpu-post-train]
extra == "tpu-post-train"
latex2sympy2-extended>=1.11.0[cuda12]
extra == "cuda12"
latex2sympy2-extended>=1.11.0[tpu]
extra == "tpu"
latex2sympy2-extended>=1.11.0[tpu-post-train]
extra == "tpu-post-train"
libclang>=18.1.1[cuda12]
extra == "cuda12"
libclang>=18.1.1[runner]
extra == "runner"
libclang>=18.1.1[tpu]
extra == "tpu"
libclang>=18.1.1[tpu-post-train]
extra == "tpu-post-train"
libcst>=1.9.0[cuda12]
extra == "cuda12"
libcst>=1.9.0[tpu]
extra == "tpu"
libcst>=1.9.0[tpu-post-train]
extra == "tpu-post-train"
libtpu>=0.0.30[runner]
(platform_machine == "x86_64" and sys_platform == "linux") and extra == "runner"
libtpu>=0.0.46[tpu]
(platform_machine == "x86_64" and sys_platform == "linux") and extra == "tpu"
libtpu>=0.0.44[tpu-post-train]
extra == "tpu-post-train"
llguidance>=1.7.6[tpu-post-train]
extra == "tpu-post-train"
llvmlite>=0.49.0[tpu-post-train]
extra == "tpu-post-train"
lm-format-enforcer>=0.11.3[tpu-post-train]
extra == "tpu-post-train"
loguru>=0.7.3[tpu-post-train]
extra == "tpu-post-train"
lxml>=6.1.3[cuda12]
extra == "cuda12"
lxml>=6.1.3[tpu]
extra == "tpu"
lxml>=6.1.3[tpu-post-train]
extra == "tpu-post-train"
markdown>=3.10.2[cuda12]
extra == "cuda12"
markdown>=3.10[runner]
extra == "runner"
markdown>=3.10.2[tpu]
extra == "tpu"
markdown>=3.4.1[tpu-post-train]
extra == "tpu-post-train"
markdown-it-py>=4.2.0[cuda12]
extra == "cuda12"
markdown-it-py>=4.0.0[runner]
extra == "runner"
markdown-it-py>=4.2.0[tpu]
extra == "tpu"
markdown-it-py>=4.2.0[tpu-post-train]
extra == "tpu-post-train"
markupsafe>=3.0.3[cuda12]
extra == "cuda12"
markupsafe>=3.0.3[runner]
extra == "runner"
markupsafe>=3.0.3[tpu]
extra == "tpu"
markupsafe>=3.0.3[tpu-post-train]
extra == "tpu-post-train"
math-verify>=0.9.0[cuda12]
extra == "cuda12"
math-verify>=0.9.0[tpu]
extra == "tpu"
math-verify>=0.9.0[tpu-post-train]
extra == "tpu-post-train"
matplotlib>=3.11.1[cuda12]
extra == "cuda12"
matplotlib>=3.10.7[runner]
extra == "runner"
matplotlib>=3.11.1[tpu]
extra == "tpu"
matplotlib>=3.11.0[tpu-post-train]
extra == "tpu-post-train"
matplotlib-inline>=0.2.2[tpu-post-train]
extra == "tpu-post-train"
mccabe>=0.7.0[cuda12]
extra == "cuda12"
mccabe>=0.7.0[tpu]
extra == "tpu"
mccabe>=0.7.0[tpu-post-train]
extra == "tpu-post-train"
mcp>=1.26.0[tpu-post-train]
extra == "tpu-post-train"
mdurl>=0.1.2[cuda12]
extra == "cuda12"
mdurl>=0.1.2[runner]
extra == "runner"
mdurl>=0.1.2[tpu]
extra == "tpu"
mdurl>=0.1.2[tpu-post-train]
extra == "tpu-post-train"
mistral-common>=1.11.7[tpu-post-train]
extra == "tpu-post-train"
ml-collections>=1.1.0[cuda12]
extra == "cuda12"
ml-collections>=1.1.0[tpu]
extra == "tpu"
ml-collections>=1.1.0[tpu-post-train]
extra == "tpu-post-train"
ml-dtypes>=0.6.0[cuda12]
extra == "cuda12"
ml-dtypes>=0.5.4[runner]
extra == "runner"
ml-dtypes>=0.6.0[tpu]
extra == "tpu"
ml-dtypes>=0.5.4[tpu-post-train]
extra == "tpu-post-train"
ml-goodput-measurement>=0.2.3[cuda12]
extra == "cuda12"
ml-goodput-measurement>=0.2.3[tpu]
extra == "tpu"
ml-goodput-measurement>=0.2.3[tpu-post-train]
extra == "tpu-post-train"
model-hosting-container-standards>=0.1.16[tpu-post-train]
extra == "tpu-post-train"
more-itertools>=11.1.0[cuda12]
extra == "cuda12"
more-itertools>=11.1.0[tpu]
extra == "tpu"
more-itertools>=11.1.0[tpu-post-train]
extra == "tpu-post-train"
mpmath>=1.4.1[cuda12]
extra == "cuda12"
mpmath>=1.3.0[runner]
extra == "runner"
mpmath>=1.4.1[tpu]
extra == "tpu"
mpmath>=1.3.0[tpu-post-train]
extra == "tpu-post-train"
msgpack>=1.2.2[cuda12]
extra == "cuda12"
msgpack>=1.2.2[tpu]
extra == "tpu"
msgpack>=1.2.2[tpu-post-train]
extra == "tpu-post-train"
msgspec>=0.21.1[cuda12]
extra == "cuda12"
msgspec>=0.21.1[tpu]
extra == "tpu"
msgspec>=0.21.1[tpu-post-train]
extra == "tpu-post-train"
multidict>=6.9.1[cuda12]
extra == "cuda12"
multidict>=6.9.1[tpu]
extra == "tpu"
multidict>=6.9.1[tpu-post-train]
extra == "tpu-post-train"
multiprocess>=0.70.19[cuda12]
extra == "cuda12"
multiprocess>=0.70.19[tpu]
extra == "tpu"
multiprocess>=0.70.19[tpu-post-train]
extra == "tpu-post-train"
mypy-extensions>=1.1.0[cuda12]
extra == "cuda12"
mypy-extensions>=1.1.0[tpu]
extra == "tpu"
mypy-extensions>=1.1.0[tpu-post-train]
extra == "tpu-post-train"
myst-nb[docs]
extra == "docs"
myst-parser[docs]
extra == "docs"
namex>=0.1.0[cuda12]
extra == "cuda12"
namex>=0.1.0[runner]
extra == "runner"
namex>=0.1.0[tpu]
extra == "tpu"
namex>=0.1.0[tpu-post-train]
extra == "tpu-post-train"
nbclient>=0.11.0[tpu-post-train]
extra == "tpu-post-train"
nbformat>=5.11.1[tpu-post-train]
extra == "tpu-post-train"
nest-asyncio>=1.6.0[cuda12]
sys_platform == "win32" and extra == "cuda12"
nest-asyncio>=1.6.0[tpu]
sys_platform == "win32" and extra == "tpu"
nest-asyncio>=1.6.0[tpu-post-train]
sys_platform == "win32" and extra == "tpu-post-train"
nest-asyncio2>=1.7.2[tpu-post-train]
extra == "tpu-post-train"
networkx>=3.7[cuda12]
extra == "cuda12"
networkx>=3.7[tpu]
extra == "tpu"
networkx>=3.7[tpu-post-train]
extra == "tpu-post-train"
ninja>=1.13.2[cuda12]
extra == "cuda12"
ninja>=1.13.2[tpu]
extra == "tpu"
ninja>=1.13.2[tpu-post-train]
extra == "tpu-post-train"
nltk>=3.10.3[cuda12]
extra == "cuda12"
nltk>=3.10.3[tpu]
extra == "tpu"
nltk>=3.10.3[tpu-post-train]
extra == "tpu-post-train"
nodeenv>=1.10.0[cuda12]
extra == "cuda12"
nodeenv>=1.10.0[tpu]
extra == "tpu"
nodeenv>=1.10.0[tpu-post-train]
extra == "tpu-post-train"
numba>=0.67.0[tpu-post-train]
extra == "tpu-post-train"
numpy>=2.1.3[cuda12]
extra == "cuda12"
numpy>=2.0.2[runner]
extra == "runner"
numpy>=2.1.3[tpu]
extra == "tpu"
numpy>=2.1.3[tpu-post-train]
extra == "tpu-post-train"
numpy-typing-compat>=20260602.2.1[cuda12]
extra == "cuda12"
numpy-typing-compat>=20250818.2.0[runner]
extra == "runner"
numpy-typing-compat>=20260602.2.1[tpu]
extra == "tpu"
numpy-typing-compat>=20260602.2.1[tpu-post-train]
extra == "tpu-post-train"
nvidia-cublas-cu12>=12.9.2.10[cuda12]
sys_platform == "linux" and extra == "cuda12"
nvidia-cuda-cccl>=13.3.3.4.1[cuda12]
sys_platform == "linux" and extra == "cuda12"
nvidia-cuda-cccl>=13.3.3.4.1[tpu]
sys_platform == "linux" and extra == "tpu"
nvidia-cuda-cccl-cu12>=12.9.27[cuda12]
sys_platform == "linux" and extra == "cuda12"
nvidia-cuda-cupti-cu12>=12.9.79[cuda12]
sys_platform == "linux" and extra == "cuda12"
nvidia-cuda-nvcc-cu12>=12.9.86[cuda12]
sys_platform == "linux" and extra == "cuda12"
nvidia-cuda-nvrtc-cu12>=12.9.86[cuda12]
sys_platform == "linux" and extra == "cuda12"
nvidia-cuda-runtime-cu12>=12.9.79[cuda12]
sys_platform == "linux" and extra == "cuda12"
nvidia-cudnn-cu12>=9.24.0.43[cuda12]
sys_platform == "linux" and extra == "cuda12"
nvidia-cudnn-frontend>=1.29.0[cuda12]
extra == "cuda12"
nvidia-cufft-cu12>=11.4.1.4[cuda12]
sys_platform == "linux" and extra == "cuda12"
nvidia-cusolver-cu12>=11.7.5.82[cuda12]
sys_platform == "linux" and extra == "cuda12"
nvidia-cusparse-cu12>=12.5.10.65[cuda12]
sys_platform == "linux" and extra == "cuda12"
nvidia-cutlass-dsl>=4.8.0[cuda12]
extra == "cuda12"
nvidia-cutlass-dsl-libs-base>=4.8.0[cuda12]
extra == "cuda12"
nvidia-cutlass-dsl-libs-core>=4.8.0[cuda12]
extra == "cuda12"
nvidia-cutlass-dsl-libs-cu12>=4.8.0[cuda12]
extra == "cuda12"
nvidia-cutlass-dsl-libs-cu13>=4.8.0[cuda12]
extra == "cuda12"
nvidia-ml-py>=13.610.43[cuda12]
extra == "cuda12"
nvidia-ml-py>=13.610.43[tpu]
extra == "tpu"
nvidia-nccl-cu12>=2.31.2[cuda12]
sys_platform == "linux" and extra == "cuda12"
nvidia-nvjitlink-cu12>=12.9.86[cuda12]
sys_platform == "linux" and extra == "cuda12"
nvidia-nvshmem-cu12>=3.7.2[cuda12]
sys_platform == "linux" and extra == "cuda12"
oauthlib>=3.3.1[cuda12]
extra == "cuda12"
oauthlib>=3.3.1[runner]
extra == "runner"
oauthlib>=3.3.1[tpu]
extra == "tpu"
oauthlib>=3.3.1[tpu-post-train]
extra == "tpu-post-train"
omegaconf>=2.3.1[cuda12]
extra == "cuda12"
omegaconf>=2.3.0[runner]
extra == "runner"
omegaconf>=2.3.1[tpu]
extra == "tpu"
omegaconf>=2.3.1[tpu-post-train]
extra == "tpu-post-train"
openai>=3.16.2[tpu-post-train]
extra == "tpu-post-train"
openai-harmony>=0.0.8[tpu-post-train]
extra == "tpu-post-train"
opencensus>=0.0.1[tpu-post-train]
extra == "tpu-post-train"
opencv-python-headless>=4.13.0.90[tpu-post-train]
extra == "tpu-post-train"
opentelemetry-api>=1.44.0[cuda12]
extra == "cuda12"
opentelemetry-api>=1.44.0[tpu]
extra == "tpu"
opentelemetry-api>=1.44.0[tpu-post-train]
extra == "tpu-post-train"
opentelemetry-exporter-otlp>=1.44.0[tpu-post-train]
extra == "tpu-post-train"
opentelemetry-exporter-otlp-proto-common>=1.44.0[tpu-post-train]
extra == "tpu-post-train"
opentelemetry-exporter-otlp-proto-grpc>=1.44.0[tpu-post-train]
extra == "tpu-post-train"
opentelemetry-exporter-otlp-proto-http>=1.44.0[tpu-post-train]
extra == "tpu-post-train"
opentelemetry-exporter-prometheus>=0.29b0[tpu-post-train]
extra == "tpu-post-train"
opentelemetry-proto>=1.44.0[tpu-post-train]
extra == "tpu-post-train"
opentelemetry-sdk>=1.44.0[tpu-post-train]
extra == "tpu-post-train"
opentelemetry-semantic-conventions>=0.65b0[tpu-post-train]
extra == "tpu-post-train"
opentelemetry-semantic-conventions-ai>=0.4.1[tpu-post-train]
extra == "tpu-post-train"
opt-einsum>=3.4.0[cuda12]
extra == "cuda12"
opt-einsum>=3.4.0[runner]
extra == "runner"
opt-einsum>=3.4.0[tpu]
extra == "tpu"
opt-einsum>=3.4.0[tpu-post-train]
extra == "tpu-post-train"
optax>=0.2.8[cuda12]
extra == "cuda12"
optax>=0.2.8[tpu]
extra == "tpu"
optax>=0.2.6[tpu-post-train]
extra == "tpu-post-train"
optree>=0.19.1[cuda12]
extra == "cuda12"
optree>=0.18.0[runner]
extra == "runner"
optree>=0.19.1[tpu]
extra == "tpu"
optree>=0.19.1[tpu-post-train]
extra == "tpu-post-train"
optype>=0.18.0[cuda12]
extra == "cuda12"
optype>=0.14.0[runner]
extra == "runner"
optype>=0.18.0[tpu]
extra == "tpu"
optype>=0.18.0[tpu-post-train]
extra == "tpu-post-train"
orbax-checkpoint>=0.12.4[cuda12]
extra == "cuda12"
orbax-checkpoint>=0.12.4[tpu]
extra == "tpu"
orbax-checkpoint>=0.12.4[tpu-post-train]
extra == "tpu-post-train"
outlines-core>=0.2.14[tpu-post-train]
extra == "tpu-post-train"
packaging>=26.3[cuda12]
extra == "cuda12"
packaging>=24.2[runner]
extra == "runner"
packaging>=26.3[tpu]
extra == "tpu"
packaging>=26.2[tpu-post-train]
extra == "tpu-post-train"
pandas>=3.0.6[cuda12]
extra == "cuda12"
pandas>=3.0.6[tpu]
extra == "tpu"
pandas>=3.0.6[tpu-post-train]
extra == "tpu-post-train"
papermill>=2.7.0[tpu-post-train]
extra == "tpu-post-train"
parameterized>=0.9.0[cuda12]
extra == "cuda12"
parameterized>=0.9.0[tpu]
extra == "tpu"
parameterized>=0.9.0[tpu-post-train]
extra == "tpu-post-train"
parso>=0.8.7[tpu-post-train]
extra == "tpu-post-train"
partial-json-parser>=0.2.1.1.post7[tpu-post-train]
extra == "tpu-post-train"
pathspec>=1.1.1[cuda12]
extra == "cuda12"
pathspec>=1.1.1[tpu]
extra == "tpu"
pathspec>=1.1.1[tpu-post-train]
extra == "tpu-post-train"
pathwaysutils>=0.1.11[cuda12]
extra == "cuda12"
pathwaysutils>=0.1.11[tpu]
extra == "tpu"
pathwaysutils>=0.1.11[tpu-post-train]
extra == "tpu-post-train"
peft>=0.21.0[tpu-post-train]
extra == "tpu-post-train"
perfetto>=0.58.2[tpu-post-train]
extra == "tpu-post-train"
pexpect>=4.9.0[tpu-post-train]
(sys_platform != "emscripten" and sys_platform != "win32") and extra == "tpu-post-train"
pillow>=12.3.0[cuda12]
extra == "cuda12"
pillow>=12.0.0[runner]
extra == "runner"
pillow>=12.3.0[tpu]
extra == "tpu"
pillow>=12.3.0[tpu-post-train]
extra == "tpu-post-train"
platformdirs>=4.11.11[cuda12]
extra == "cuda12"
platformdirs>=4.11.11[tpu]
extra == "tpu"
platformdirs>=4.11.11[tpu-post-train]
extra == "tpu-post-train"
pluggy>=1.6.0[cuda12]
extra == "cuda12"
pluggy>=1.6.0[runner]
extra == "runner"
pluggy>=1.6.0[tpu]
extra == "tpu"
pluggy>=1.6.0[tpu-post-train]
extra == "tpu-post-train"
portpicker>=1.6.0[cuda12]
extra == "cuda12"
portpicker>=1.6.0[runner]
extra == "runner"
portpicker>=1.6.0[tpu]
extra == "tpu"
portpicker>=1.6.0[tpu-post-train]
extra == "tpu-post-train"
pre-commit>=4.6.2[cuda12]
extra == "cuda12"
pre-commit>=4.6.2[tpu]
extra == "tpu"
pre-commit>=4.6.2[tpu-post-train]
extra == "tpu-post-train"
prometheus-client>=0.26.0[cuda12]
extra == "cuda12"
prometheus-client>=0.26.0[tpu]
extra == "tpu"
prometheus-client>=0.26.0[tpu-post-train]
extra == "tpu-post-train"
prometheus-fastapi-instrumentator>=8.1.0[tpu-post-train]
extra == "tpu-post-train"
promise>=2.3[tpu-post-train]
extra == "tpu-post-train"
prompt-toolkit>=3.0.53[tpu-post-train]
extra == "tpu-post-train"
propcache>=0.5.4[cuda12]
extra == "cuda12"
propcache>=0.5.4[tpu]
extra == "tpu"
propcache>=0.5.4[tpu-post-train]
extra == "tpu-post-train"
proto-plus>=1.28.4[cuda12]
extra == "cuda12"
proto-plus>=1.27.1[runner]
extra == "runner"
proto-plus>=1.28.4[tpu]
extra == "tpu"
proto-plus>=1.28.4[tpu-post-train]
extra == "tpu-post-train"
protobuf>=6.33.6[cuda12]
extra == "cuda12"
protobuf>=5.29.6[runner]
extra == "runner"
protobuf>=6.33.6[tpu]
extra == "tpu"
protobuf>=6.33.6[tpu-post-train]
extra == "tpu-post-train"
psutil>=7.2.2[cuda12]
extra == "cuda12"
psutil>=7.1.3[runner]
extra == "runner"
psutil>=7.2.2[tpu]
extra == "tpu"
psutil>=7.2.2[tpu-post-train]
extra == "tpu-post-train"
ptyprocess>=0.7.0[tpu-post-train]
(sys_platform != "emscripten" and sys_platform != "win32") and extra == "tpu-post-train"
pure-eval>=0.2.4[tpu-post-train]
extra == "tpu-post-train"
py-cpuinfo>=9.0.0[tpu-post-train]
extra == "tpu-post-train"
py-spy>=0.4.0[tpu-post-train]
extra == "tpu-post-train"
pyarrow>=25.0.1[cuda12]
extra == "cuda12"
pyarrow>=25.0.1[tpu]
extra == "tpu"
pyarrow>=25.0.1[tpu-post-train]
extra == "tpu-post-train"
pyasn1>=0.6.4[cuda12]
extra == "cuda12"
pyasn1>=0.6.3[runner]
extra == "runner"
pyasn1>=0.6.4[tpu]
extra == "tpu"
pyasn1>=0.6.4[tpu-post-train]
extra == "tpu-post-train"
pyasn1-modules>=0.4.2[cuda12]
extra == "cuda12"
pyasn1-modules>=0.4.2[runner]
extra == "runner"
pyasn1-modules>=0.4.2[tpu]
extra == "tpu"
pyasn1-modules>=0.4.2[tpu-post-train]
extra == "tpu-post-train"
pybase64>=1.5.0[tpu-post-train]
extra == "tpu-post-train"
pycnite>=2024.7.31[cuda12]
extra == "cuda12"
pycnite>=2024.7.31[tpu]
extra == "tpu"
pycnite>=2024.7.31[tpu-post-train]
extra == "tpu-post-train"
pycountry>=26.2.16[tpu-post-train]
extra == "tpu-post-train"
pycparser>=3.0[cuda12]
(implementation_name != "PyPy" and platform_python_implementation != "PyPy") and extra == "cuda12"
pycparser>=3.0[runner]
(implementation_name != "PyPy" and platform_python_implementation != "PyPy") and extra == "runner"
pycparser>=3.0[tpu]
(implementation_name != "PyPy" and platform_python_implementation != "PyPy") and extra == "tpu"
pycparser>=3.0[tpu-post-train]
((implementation_name != "PyPy" and platform_python_implementation != "PyPy") or (implementation_name == "pypy" and platform_python_implementation == "PyPy")) and extra == "tpu-post-train"
pycryptodomex>=3.23.0[cuda12]
extra == "cuda12"
pycryptodomex>=3.23.0[tpu]
extra == "tpu"
pycryptodomex>=3.23.0[tpu-post-train]
extra == "tpu-post-train"
pydantic>=2.13.5[cuda12]
extra == "cuda12"
pydantic>=2.12.5[runner]
extra == "runner"
pydantic>=2.13.5[tpu]
extra == "tpu"
pydantic>=2.13.5[tpu-post-train]
extra == "tpu-post-train"
pydantic-core>=2.46.5[cuda12]
extra == "cuda12"
pydantic-core>=2.41.5[runner]
extra == "runner"
pydantic-core>=2.46.5[tpu]
extra == "tpu"
pydantic-core>=2.46.5[tpu-post-train]
extra == "tpu-post-train"
pydantic-extra-types>=2.11.1[tpu-post-train]
extra == "tpu-post-train"
pydantic-settings>=2.5.2[tpu-post-train]
extra == "tpu-post-train"
pydot>=4.0.1[cuda12]
extra == "cuda12"
pydot>=4.0.1[tpu]
extra == "tpu"
pydot>=4.0.1[tpu-post-train]
extra == "tpu-post-train"
pyelftools>=0.33[cuda12]
extra == "cuda12"
pyelftools>=0.32[runner]
extra == "runner"
pyelftools>=0.33[tpu]
extra == "tpu"
pyelftools>=0.33[tpu-post-train]
extra == "tpu-post-train"
pygments>=2.21.0[cuda12]
extra == "cuda12"
pygments>=2.19.2[runner]
extra == "runner"
pygments>=2.21.0[tpu]
extra == "tpu"
pygments>=2.20.0[tpu-post-train]
extra == "tpu-post-train"
pyink>=26.5.1[cuda12]
extra == "cuda12"
pyink>=26.5.1[tpu]
extra == "tpu"
pyink>=26.5.1[tpu-post-train]
extra == "tpu-post-train"
pyjwt>=2.10.1[tpu-post-train]
extra == "tpu-post-train"
pylatexenc>=2.11[tpu-post-train]
extra == "tpu-post-train"
pylint>=4.0.8[cuda12]
extra == "cuda12"
pylint>=4.0.8[tpu]
extra == "tpu"
pylint>=4.0.8[tpu-post-train]
extra == "tpu-post-train"
pynvml>=13.0.1[cuda12]
extra == "cuda12"
pynvml>=13.0.1[tpu]
extra == "tpu"
pynvml>=13.0.1[tpu-post-train]
extra == "tpu-post-train"
pyparsing>=3.3.2[cuda12]
extra == "cuda12"
pyparsing>=3.2.5[runner]
extra == "runner"
pyparsing>=3.3.2[tpu]
extra == "tpu"
pyparsing>=3.3.2[tpu-post-train]
extra == "tpu-post-train"
pyproject-hooks>=1.2.0[cuda12]
extra == "cuda12"
pyproject-hooks>=1.2.0[runner]
extra == "runner"
pyproject-hooks>=1.2.0[tpu]
extra == "tpu"
pyproject-hooks>=1.2.0[tpu-post-train]
extra == "tpu-post-train"
pytest>=9.1.1[cuda12]
extra == "cuda12"
pytest>=8.4.2[runner]
extra == "runner"
pytest>=9.1.1[tpu]
extra == "tpu"
pytest>=9.1.1[tpu-post-train]
extra == "tpu-post-train"
pytest-mock>=3.15.1[tpu-post-train]
extra == "tpu-post-train"
pytest-xdist>=3.8.0[cuda12]
extra == "cuda12"
pytest-xdist>=3.8.0[runner]
extra == "runner"
pytest-xdist>=3.8.0[tpu]
extra == "tpu"
pytest-xdist>=3.8.0[tpu-post-train]
extra == "tpu-post-train"
python-dateutil>=2.9.0.post0[cuda12]
extra == "cuda12"
python-dateutil>=2.9.0.post0[runner]
extra == "runner"
python-dateutil>=2.9.0.post0[tpu]
extra == "tpu"
python-dateutil>=2.9.0.post0[tpu-post-train]
extra == "tpu-post-train"
python-discovery>=1.6.1[cuda12]
extra == "cuda12"
python-discovery>=1.6.1[tpu]
extra == "tpu"
python-discovery>=1.6.1[tpu-post-train]
extra == "tpu-post-train"
python-dotenv>=1.2.3[tpu-post-train]
extra == "tpu-post-train"
python-json-logger>=4.2.0[tpu-post-train]
extra == "tpu-post-train"
python-multipart>=0.0.9[tpu-post-train]
extra == "tpu-post-train"
pytokens>=0.4.1[cuda12]
extra == "cuda12"
pytokens>=0.4.1[tpu]
extra == "tpu"
pytokens>=0.4.1[tpu-post-train]
extra == "tpu-post-train"
pytype>=2024.10.11[cuda12]
extra == "cuda12"
pytype>=2024.10.11[tpu]
extra == "tpu"
pytype>=2024.10.11[tpu-post-train]
extra == "tpu-post-train"
pywin32>=311[runner]
sys_platform == "win32" and extra == "runner"
pywin32>=310[tpu-post-train]
sys_platform == "win32" and extra == "tpu-post-train"
pyyaml>=6.0.3[cuda12]
extra == "cuda12"
pyyaml>=6.0.2[runner]
extra == "runner"
pyyaml>=6.0.3[tpu]
extra == "tpu"
pyyaml>=6.0.3[tpu-post-train]
extra == "tpu-post-train"
pyzmq>=27.2.0[tpu-post-train]
extra == "tpu-post-train"
qwix>=0.1.8[cuda12]
extra == "cuda12"
qwix>=0.1.8[tpu]
extra == "tpu"
qwix>=0.1.8[tpu-post-train]
extra == "tpu-post-train"
ray>=2.58.0[tpu-post-train]
extra == "tpu-post-train"
referencing>=0.37.0[tpu-post-train]
extra == "tpu-post-train"
regex>=2026.9.10[cuda12]
extra == "cuda12"
regex>=2026.9.10[tpu]
extra == "tpu"
regex>=2026.9.10[tpu-post-train]
extra == "tpu-post-train"
requests>=2.34.2[cuda12]
extra == "cuda12"
requests>=2.32.5[runner]
extra == "runner"
requests>=2.34.2[tpu]
extra == "tpu"
requests>=2.34.2[tpu-post-train]
extra == "tpu-post-train"
requests-oauthlib>=2.0.0[cuda12]
extra == "cuda12"
requests-oauthlib>=2.0.0[runner]
extra == "runner"
requests-oauthlib>=2.0.0[tpu]
extra == "tpu"
requests-oauthlib>=2.0.0[tpu-post-train]
extra == "tpu-post-train"
responses>=0.18.0[cuda12]
extra == "cuda12"
responses>=0.18.0[tpu]
extra == "tpu"
rich>=15.0.0[cuda12]
extra == "cuda12"
rich>=14.2.0[runner]
extra == "runner"
rich>=15.0.0[tpu]
extra == "tpu"
rich>=15.0.0[tpu-post-train]
extra == "tpu-post-train"
rpds-py>=2026.6.3[tpu-post-train]
extra == "tpu-post-train"
ruamel-yaml>=0.18.10[runner]
extra == "runner"
ruamel-yaml-clib>=0.2.15[runner]
platform_python_implementation == "CPython" and extra == "runner"
runai-model-streamer>=0.15.9[tpu-post-train]
extra == "tpu-post-train"
runai-model-streamer-gcs>=0.15.9[tpu-post-train]
extra == "tpu-post-train"
runai-model-streamer-s3>=0.15.9[tpu-post-train]
extra == "tpu-post-train"
s3transfer>=0.9.0[tpu-post-train]
extra == "tpu-post-train"
safetensors>=0.8.0[cuda12]
extra == "cuda12"
safetensors>=0.8.0[tpu]
extra == "tpu"
safetensors>=0.7.0[tpu-post-train]
extra == "tpu-post-train"
scipy>=1.18.0[cuda12]
extra == "cuda12"
scipy>=1.16.3[runner]
extra == "runner"
scipy>=1.18.0[tpu]
extra == "tpu"
scipy>=1.18.0[tpu-post-train]
extra == "tpu-post-train"
scipy-stubs>=1.18.0.1[cuda12]
extra == "cuda12"
scipy-stubs>=1.16.3.0[runner]
extra == "runner"
scipy-stubs>=1.18.0.1[tpu]
extra == "tpu"
scipy-stubs>=1.18.0.1[tpu-post-train]
extra == "tpu-post-train"
semantic-version>=2.8.2[tpu-post-train]
extra == "tpu-post-train"
sentencepiece>=0.2.2[cuda12]
extra == "cuda12"
sentencepiece>=0.2.2[tpu]
extra == "tpu"
sentencepiece>=0.2.2[tpu-post-train]
extra == "tpu-post-train"
setproctitle>=1.3.7[tpu-post-train]
extra == "tpu-post-train"
setuptools>=84.0.0[cuda12]
extra == "cuda12"
setuptools>=80.9.0[runner]
extra == "runner"
setuptools>=84.0.0[tpu]
extra == "tpu"
setuptools>=78.1.0[tpu-post-train]
extra == "tpu-post-train"
setuptools-rust>=1.9.0[tpu-post-train]
extra == "tpu-post-train"
setuptools-scm>=9.2.2[runner]
extra == "runner"
setuptools-scm>=8.0.1[tpu-post-train]
extra == "tpu-post-train"
shellingham>=1.5.4[cuda12]
extra == "cuda12"
shellingham>=1.5.4[tpu]
extra == "tpu"
shellingham>=1.5.4[tpu-post-train]
extra == "tpu-post-train"
shortuuid>=1.0.13[cuda12]
extra == "cuda12"
shortuuid>=1.0.13[tpu]
extra == "tpu"
shortuuid>=1.0.13[tpu-post-train]
extra == "tpu-post-train"
simple-parsing>=0.1.9[tpu-post-train]
extra == "tpu-post-train"
simplejson>=4.1.2[cuda12]
extra == "cuda12"
simplejson>=4.1.2[tpu]
extra == "tpu"
simplejson>=4.1.2[tpu-post-train]
extra == "tpu-post-train"
six>=1.17.0[cuda12]
extra == "cuda12"
six>=1.17.0[runner]
extra == "runner"
six>=1.17.0[tpu]
extra == "tpu"
six>=1.17.0[tpu-post-train]
extra == "tpu-post-train"
smart-open>=0.1.0[tpu-post-train]
extra == "tpu-post-train"
sniffio>=1.3.1[cuda12]
extra == "cuda12"
sniffio>=1.3.1[runner]
extra == "runner"
sniffio>=1.3.1[tpu]
extra == "tpu"
sniffio>=1.3.1[tpu-post-train]
extra == "tpu-post-train"
sortedcontainers>=2.4.0[cuda12]
extra == "cuda12"
sortedcontainers>=2.4.0[runner]
extra == "runner"
sortedcontainers>=2.4.0[tpu]
extra == "tpu"
sortedcontainers>=2.4.0[tpu-post-train]
extra == "tpu-post-train"
sphinx<9[docs]
extra == "docs"
sphinx-book-theme[docs]
extra == "docs"
sphinx-copybutton[docs]
extra == "docs"
sphinx-design[docs]
extra == "docs"
sse-starlette>=1.6.1[tpu-post-train]
extra == "tpu-post-train"
stack-data>=0.6.3[tpu-post-train]
extra == "tpu-post-train"
starlette>=1.6.0[cuda12]
extra == "cuda12"
starlette>=1.6.0[tpu]
extra == "tpu"
starlette>=1.6.0[tpu-post-train]
extra == "tpu-post-train"
supervisor>=4.3.0[tpu-post-train]
extra == "tpu-post-train"
sympy>=1.12[cuda12]
extra == "cuda12"
sympy>=1.12[tpu]
extra == "tpu"
sympy>=1.14.0[tpu-post-train]
extra == "tpu-post-train"
tabulate>=0.10.0[cuda12]
extra == "cuda12"
tabulate>=0.9.0[runner]
extra == "runner"
tabulate>=0.10.0[tpu]
extra == "tpu"
tabulate>=0.10.0[tpu-post-train]
extra == "tpu-post-train"
tenacity>=9.1.4[cuda12]
extra == "cuda12"
tenacity>=9.1.4[runner]
extra == "runner"
tenacity>=9.1.4[tpu]
extra == "tpu"
tenacity>=9.1.4[tpu-post-train]
extra == "tpu-post-train"
tensorboard-data-server>=0.7.2[runner]
extra == "runner"
tensorboard-plugin-profile>=2.13.0[cuda12]
extra == "cuda12"
tensorboard-plugin-profile>=2.13.0[runner]
extra == "runner"
tensorboard-plugin-profile>=2.13.0[tpu]
extra == "tpu"
tensorboard-plugin-profile>=2.13.0[tpu-post-train]
extra == "tpu-post-train"
tensorboardx>=2.6.5[cuda12]
extra == "cuda12"
tensorboardx>=2.6.5[tpu]
extra == "tpu"
tensorboardx>=2.6.5[tpu-post-train]
extra == "tpu-post-train"
tensorflow-datasets>=4.9.10[tpu-post-train]
extra == "tpu-post-train"
tensorflow-metadata>=1.21.0[tpu-post-train]
extra == "tpu-post-train"
tensorstore>=0.1.85[cuda12]
extra == "cuda12"
tensorstore>=0.1.79[runner]
extra == "runner"
tensorstore>=0.1.85[tpu]
extra == "tpu"
tensorstore>=0.1.84[tpu-post-train]
extra == "tpu-post-train"
termcolor>=3.3.0[cuda12]
extra == "cuda12"
termcolor>=3.2.0[runner]
extra == "runner"
termcolor>=3.3.0[tpu]
extra == "tpu"
termcolor>=3.3.0[tpu-post-train]
extra == "tpu-post-train"
tiktoken>=0.14.0[cuda12]
extra == "cuda12"
tiktoken>=0.14.0[tpu]
extra == "tpu"
tiktoken>=0.14.0[tpu-post-train]
extra == "tpu-post-train"
tokamax>=0.0.14[cuda12]
extra == "cuda12"
tokamax>=0.0.14[tpu]
extra == "tpu"
tokamax>=0.0.14[tpu-post-train]
extra == "tpu-post-train"
tokenizers>=0.23.2[cuda12]
extra == "cuda12"
tokenizers>=0.23.2[tpu]
extra == "tpu"
tokenizers>=0.22.2[tpu-post-train]
extra == "tpu-post-train"
toml>=0.10.2[cuda12]
extra == "cuda12"
toml>=0.10.2[tpu]
extra == "tpu"
toml>=0.10.2[tpu-post-train]
extra == "tpu-post-train"
tomlkit>=0.15.1[cuda12]
extra == "cuda12"
tomlkit>=0.15.1[tpu]
extra == "tpu"
tomlkit>=0.15.1[tpu-post-train]
extra == "tpu-post-train"
toolz>=1.1.0[cuda12]
extra == "cuda12"
toolz>=1.1.0[tpu]
extra == "tpu"
toolz>=1.1.0[tpu-post-train]
extra == "tpu-post-train"
torch==2.11.0+cpu[tpu-post-train]
extra == "tpu-post-train"
torchax>=0.0.13[tpu-post-train]
extra == "tpu-post-train"
torchvision==0.26.0+cpu[tpu-post-train]
extra == "tpu-post-train"
tornado>=6.5.10[tpu-post-train]
extra == "tpu-post-train"
tpu-info>=0.7.1[tpu-post-train]
extra == "tpu-post-train"
tqdm>=4.70.1[cuda12]
extra == "cuda12"
tqdm>=4.70.1[tpu]
extra == "tpu"
tqdm>=4.70.1[tpu-post-train]
extra == "tpu-post-train"
traitlets>=5.16.1[tpu-post-train]
extra == "tpu-post-train"
transformer-engine>=2.19.0[cuda12]
extra == "cuda12"
transformer-engine-cu12>=2.19.0[cuda12]
extra == "cuda12"
transformer-engine-jax>=2.19.0[cuda12]
extra == "cuda12"
transformers>=5.17.0[cuda12]
extra == "cuda12"
transformers>=5.17.0[tpu]
extra == "tpu"
transformers>=5.12.1[tpu-post-train]
extra == "tpu-post-train"
treescope>=0.1.10[cuda12]
extra == "cuda12"
treescope>=0.1.10[tpu]
extra == "tpu"
treescope>=0.1.10[tpu-post-train]
extra == "tpu-post-train"
triton>=3.6.0[tpu-post-train]
sys_platform == "linux" and extra == "tpu-post-train"
truststore>=0.10.4[tpu-post-train]
sys_platform != "emscripten" and extra == "tpu-post-train"
typeguard>=2.13.3[cuda12]
extra == "cuda12"
typeguard>=2.13.3[tpu]
extra == "tpu"
typeguard>=2.13.3[tpu-post-train]
extra == "tpu-post-train"
typer>=0.27.2[cuda12]
extra == "cuda12"
typer>=0.27.2[tpu]
extra == "tpu"
typer>=0.27.2[tpu-post-train]
extra == "tpu-post-train"
typing-extensions>=4.16.0[cuda12]
extra == "cuda12"
typing-extensions>=4.15.0[runner]
extra == "runner"
typing-extensions>=4.16.0[tpu]
extra == "tpu"
typing-extensions>=4.16.0[tpu-post-train]
extra == "tpu-post-train"
typing-inspection>=0.4.4[cuda12]
extra == "cuda12"
typing-inspection>=0.4.2[runner]
extra == "runner"
typing-inspection>=0.4.4[tpu]
extra == "tpu"
typing-inspection>=0.4.4[tpu-post-train]
extra == "tpu-post-train"
tzdata>=2026.4[cuda12]
(sys_platform == "emscripten" or sys_platform == "win32") and extra == "cuda12"
tzdata>=2026.4[tpu]
(sys_platform == "emscripten" or sys_platform == "win32") and extra == "tpu"
tzdata>=2026.4[tpu-post-train]
(sys_platform == "emscripten" or sys_platform == "win32") and extra == "tpu-post-train"
uritemplate>=4.2.0[cuda12]
extra == "cuda12"
uritemplate>=4.2.0[tpu]
extra == "tpu"
uritemplate>=4.2.0[tpu-post-train]
extra == "tpu-post-train"
urllib3>=2.7.0[cuda12]
extra == "cuda12"
urllib3>=2.5.0[runner]
extra == "runner"
urllib3>=2.7.0[tpu]
extra == "tpu"
urllib3>=2.7.0[tpu-post-train]
extra == "tpu-post-train"
uvicorn>=0.53.0[cuda12]
extra == "cuda12"
uvicorn>=0.53.0[tpu]
extra == "tpu"
uvicorn>=0.53.0[tpu-post-train]
extra == "tpu-post-train"
uvloop>=0.22.1[cuda12]
extra == "cuda12"
uvloop>=0.22.1[tpu]
extra == "tpu"
uvloop>=0.22.1[tpu-post-train]
extra == "tpu-post-train"
virtualenv>=21.9.0[cuda12]
extra == "cuda12"
virtualenv>=21.9.0[tpu]
extra == "tpu"
virtualenv>=21.9.0[tpu-post-train]
extra == "tpu-post-train"
wadler-lindig>=0.1.7[cuda12]
extra == "cuda12"
wadler-lindig>=0.1.7[tpu]
extra == "tpu"
wadler-lindig>=0.1.7[tpu-post-train]
extra == "tpu-post-train"
watchfiles>=1.3.0[tpu-post-train]
extra == "tpu-post-train"
wcwidth>=0.8.4[tpu-post-train]
extra == "tpu-post-train"
websocket-client>=1.9.0[runner]
extra == "runner"
websockets>=16.1.1[cuda12]
extra == "cuda12"
websockets>=16.0[runner]
extra == "runner"
websockets>=16.1.1[tpu]
extra == "tpu"
websockets>=16.1.1[tpu-post-train]
extra == "tpu-post-train"
werkzeug>=3.1.8[cuda12]
extra == "cuda12"
werkzeug>=3.1.3[runner]
extra == "runner"
werkzeug>=3.1.8[tpu]
extra == "tpu"
werkzeug>=3.1.8[tpu-post-train]
extra == "tpu-post-train"
wheel>=0.48.0[cuda12]
extra == "cuda12"
wheel>=0.46.1[runner]
extra == "runner"
wheel>=0.48.0[tpu]
extra == "tpu"
wheel>=0.47.0[tpu-post-train]
extra == "tpu-post-train"
widgetsnbextension>=4.0.16[tpu-post-train]
extra == "tpu-post-train"
win32-setctime>=1.2.0[tpu-post-train]
sys_platform == "win32" and extra == "tpu-post-train"
wrapt>=2.4.1[tpu-post-train]
extra == "tpu-post-train"
xgrammar>=0.2.7[tpu-post-train]
extra == "tpu-post-train"
xpk>=1.8.0[runner]
extra == "runner"
xprof>=2.20.0[cuda12]
extra == "cuda12"
xprof>=2.20.0[tpu]
extra == "tpu"
xprof>=2.20.0[tpu-post-train]
extra == "tpu-post-train"
xxhash>=4.0.1[cuda12]
extra == "cuda12"
xxhash>=4.0.1[tpu]
extra == "tpu"
xxhash>=4.0.1[tpu-post-train]
extra == "tpu-post-train"
yapf>=0.43.0[tpu-post-train]
extra == "tpu-post-train"
yarl>=1.25.1[cuda12]
extra == "cuda12"
yarl>=1.25.1[tpu]
extra == "tpu"
yarl>=1.25.1[tpu-post-train]
extra == "tpu-post-train"
zipp>=4.1.0[cuda12]
extra == "cuda12"
zipp>=3.23.0[runner]
extra == "runner"
zipp>=4.1.0[tpu]
extra == "tpu"
zipp>=4.1.0[tpu-post-train]
extra == "tpu-post-train"
zstandard>=0.25.0[cuda12]
extra == "cuda12"
zstandard>=0.25.0[runner]
extra == "runner"
zstandard>=0.25.0[tpu]
extra == "tpu"
zstandard>=0.25.0[tpu-post-train]
extra == "tpu-post-train"

Used By

Not used by any tracked packages

Repository details

Forks

623

Issues

Last Commit

1 week ago

Contributors

308

Open PRs

325

Archived

GitHub Topics

Large Language Models Llm Gpt Deepseek Fine Tuning Gemma2 Gemma3 Jax Llama2 Llama3 Llama4 Mistral Mixtral Sft

Data and sources

Tracking Since

September 26, 2026

Data Updated

October 11, 2026

Data Freshness

Fresh

Keywords

deepseek gemma jax llama llm mistral mixtral

Sources and methodology

Package Information

Package Insights

Downloads by OS

Python Versions

Top Countries

Weekly Downloads

GitHub Stars

Dependencies

Used By

Repository details

Data and sources

Health Score

Maintenance

Category

Momentum

Python Support

License Type

Release Cadence

Contributor Risk

Additional Metrics