matplotlib
einops
numpy
POT 
scikit-learn
scipy
torch
torchdiffeq
torchvision
tqdm
xtb-python
openmm
mdtraj
torchdyn