rdkit
SmilesPE
torch
torchvision
