torch
pydantic>=2.11.5
tensordict
einops
ninja>=1.11.1

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