megatron-core
megatron-bridge
accelerate
fastapi
pydantic-settings
ray
ray[serve]
uvicorn
tensorstore
zarr<3.0.0,>=2.18.2
cloudpickle
fiddle
hydra-core<=1.3.2,>1.3
lightning<2.5.0
omegaconf>=2.3.0
peft
torch>=2.6.0
torchvision
torchmetrics>=0.11.0
wandb
webdataset>=0.2.86
pyparsing>2.0.2
sentencepiece
tiktoken
einops
ijson
pyarrow<21.0.0,>=20.0.0

[:platform_system != "Darwin"]
nvidia-pytriton
flashinfer-python>=0.2.5

[:platform_system != "Darwin" and platform_machine != "aarch64"]
Pillow
decord

[:sys_platform != "darwin"]
nvidia-modelopt[torch]
nvidia-resiliency-ext
transformer-engine[core_cu13,pytorch]

[inframework]

[trt-onnx]
tensorrt==10.14.1.48.post1
onnx==1.21.0
onnxscript>=0.6.0
transformers==4.51.3

[trtllm]
tensorrt-llm~=1.2.0
cuda-python~=13.0.0

[vllm]
vllm~=0.17.1
pandas
timm
