torch>=2.1.0
triton>=2.1.0

[dev]
parameterized
pandas
