datasets>=4.0.0
pillow>=11.3.0
rich>=14.1.0
safetensors>=0.6.2
tokenizers>=0.23.1
transformers<=5.16.1,>=5.6.1
typer>=0.17.4
peft==0.18.1
hf_transfer
cloudpathlib>=0.23.0
zstandard>=0.23.0
xxhash>=3.0.0

[aws]
cloudpathlib[s3]

[aws:sys_platform == "linux"]
s5cmd>=0.3.3

[azure]
cloudpathlib[azure]

[dev]
mkdocs
mkdocs-material
mkdocstrings[python]>=0.24.0
pymdown-extensions>=10.7
pytest
pytest-forked
pytest-asyncio
pre-commit
litellm
torch
ty
cloudpathlib[s3]
alembic
griffe2md

[fa4]
skyrl[megatron]

[fa4:sys_platform == "linux" and (platform_machine == "x86_64" or platform_machine == "aarch64") and python_version < "3.15"]
flash-attn-4[cu13]==4.0.0b28

[fsdp]
skyrl[skyrl-train]

[fsdp:sys_platform == "linux"]
vllm==0.30.0
vllm-router
nixl
flash-linear-attention
causal-conv1d==1.6.2.post1
flash-attn==2.8.3
torch==2.13.0
torchvision

[fsdp:sys_platform == "linux" and platform_machine == "x86_64"]
flashinfer-python==0.6.18.post1
flashinfer-jit-cache==0.6.18.post1
flashinfer-cubin==0.6.18.post1

[gcp]
cloudpathlib[gs]

[gpu]

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

[harbor]
harbor[daytona,modal]

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

[megatron]
skyrl[skyrl-train]

[megatron:sys_platform == "linux"]
transformer-engine[pytorch]==2.16.0
transformer-engine-torch==2.16.0
transformer-engine-cu13==2.16.0+cu.13.0
flash-attn==2.8.3
flash-linear-attention
causal-conv1d==1.6.2.post1
mamba-ssm==2.3.2.post1
vllm==0.30.0
vllm-router
nixl
torch==2.13.0
torchvision
megatron-bridge
megatron-core
nvidia-resiliency-ext
nvidia-modelopt
fast-hadamard-transform

[megatron:sys_platform == "linux" and platform_machine == "x86_64"]
flashinfer-python==0.6.18.post1
flashinfer-jit-cache==0.6.18.post1
flashinfer-cubin==0.6.18.post1

[miniswe]
skyrl[skyrl-train]
mini-swe-agent>=1.12.0
litellm

[mooncake]

[mooncake:sys_platform == "linux" and python_version < "3.14"]
mooncake-transfer-engine-cuda13

[ray]
ray[default]==2.57.0

[skycap]
skycap[tokens]

[skyrl-train]
loguru
tqdm
ninja
tensorboard
func_timeout
hydra-core==1.3.4
accelerate
torchdata
omegaconf
ray==2.57.0
peft==0.18.1
debugpy==1.8.0
hf_transfer
wandb
datasets>=4.0.0
tensordict
jaxtyping
skyrl-gym
polars
s3fs
gcsfs
fastapi
orjson>=3.11.9
pybase64>=1.4.2
uvicorn
pybind11
setuptools

[skyrl-train:sys_platform == "linux"]
flash-attn==2.8.3
s5cmd>=0.3.3
vllm-router
nixl

[tinker]
tinker<0.26.0,>=0.25.0
fastapi[standard]
sqlmodel
sqlalchemy[asyncio]
aiosqlite
asyncpg
psycopg2-binary

[tpu]

[tpu:sys_platform == "linux" and platform_machine == "x86_64"]
jax[tpu]>=0.7.2
