numpy
h5py
matplotlib
seaborn
torch
torchvision
pytest
