torch>=2.0.0
numpy>=1.24.0
pyyaml>=6.0
matplotlib>=3.7.0
pynvml>=11.5.0

[cuda]
ninja>=1.11.1

[dev]
pytest>=7.4.0
pytest-cov>=4.1.0
black>=23.7.0
isort>=5.12.0
flake8>=6.1.0
