datasets>=4.0.0
pillow>=11.3.0
rich>=14.1.0
safetensors>=0.6.2
tokenizers>=0.21.2
transformers<5,>=4.56.1
typer>=0.17.4
peft
hf_transfer
cloudpathlib>=0.23.0

[aws]
cloudpathlib[s3]

[azure]
cloudpathlib[azure]

[dev]
mkdocs
mkdocs-material
pytest
pytest-forked
torch
ty
cloudpathlib[s3]
alembic

[gcp]
cloudpathlib[gs]

[gpu]

[gpu:sys_platform == "linux"]
jax[cuda12]>=0.7.2

[jax]
jax<1.0,>=0.8
flax>=0.12.2
optax>=0.2.5

[jax:sys_platform == "linux"]
jax[cuda12]>=0.7.2

[skyrl_train]

[skyrl_train:python_version == "3.12" and sys_platform == "linux"]
skyrl-train[vllm]

[tinker]
tinker>=0.3.0
fastapi[standard]
sqlmodel
sqlalchemy[asyncio]
aiosqlite
asyncpg
psycopg2-binary

[tpu]

[tpu:sys_platform == "linux"]
jax[tpu]>=0.7.2
