torch>=2.0.0
psutil>=5.9.0
tensorboard>=2.11.0

[dev]
pytest>=7.0.0
pytest-cov>=4.0.0
black>=23.0.0
