torch>=2.0.0
triton>=2.0.0

[dev]
pytest>=7.0.0
pytest-benchmark>=4.0.0
pytest-cov>=4.0.0
torchvision>=0.15.0
pillow>=9.0.0
matplotlib>=3.5.0
black>=23.0.0
flake8>=6.0.0
mypy>=1.0.0
isort>=5.12.0

[examples]
torchvision>=0.15.0
pillow>=9.0.0
matplotlib>=3.5.0
