torch>=1.12.0
tensorflow>=2.10.0
pynvml>=11.0.0

[dev]
pytest
black
flake8
