torch
numpy
h5py
PyYAML
matplotlib