torch >= 1.12.0
scipy >= 1.6.0
triton; platform_system == "Linux" and platform_machine == "x86_64"
