numpy
PyYAML
torch
