torch
torchdiffeq
scipy
numpy
matplotlib
tqdm
imageio
