torch>=1.12.0
scipy>=1.6.0

[:platform_system == "Linux" and platform_machine == "x86_64"]
triton
