torch>=2.0.0
numpy
matplotlib
jupyter

[dev]
pytest
black
flake8
