numpy
pytest
tensorflow
torch
