torch
pytest
black
numpy
