torch
transformers
pytorch-fast-transformers
rdkit
numpy
