torch
pydantic>=2.11.5
tensordict
einops
ninja>=1.11.1
transformers>=4.57.0

[:platform_system == "Linux"]
triton>=2.0.0

[profile]
nvidia-ml-py
