torch
pytorch-ignite
molgrid
numpy
