torch>=2.0

[dev]
pytest>=7

[nvidia]
pynvml>=11.5.0

[viz]
matplotlib>=3.7
