fastapi
uvicorn
transformers!=4.52.*,!=4.53.*,<5.0.0,>=4.51.1
datasets>=4.4.1
pydantic
math_verify
nest_asyncio
pybind11[global]
torch>=2.6.0
torchvision>=0.21.0
torchao>=0.10.0
tensordict==0.7.2
toml
StrEnum
redis
msgpack
qwen_vl_utils
pynvml
boto3
modelscope
cloudpickle
click
rich
pyyaml
blobfile
nvidia-nccl-cu12>=2.26.2
liger_kernel
accelerate
kernels
diffusers>=0.36.0
peft
zmq>=0.0.0
nvidia-ml-py>=13.590.48

[all]
cosmos_rl[rl,vla,wfm]

[rl]
vllm>=0.8.5
flash-attn>=2.8.3

[vla]
bddl
easydict
einops
future
gym
gymnasium
hydra-core
libero
math-verify
matplotlib
numpy<2
opencv-python
robomimic
robosuite==1.4
thop
timm<1
wandb>0.13.1
transformers<5.0.0
sentencepiece

[wfm]
termcolor
aioboto3
aiobotocore
imageio
imageio-ffmpeg
webdataset
decord
trimesh
opencv-python-headless
better_profanity
nltk
retinaface-py
