torch>=2.1
triton>=3.0.0

[all]
flash-colreduce[test]
flash-colreduce[dev]
flash-colreduce[bench]

[bench]
rich>=13.7
tqdm>=4.67

[dev]
black>=24.0

[test]
pytest>=7
pytest-cov>=4
