matplotlib
numpy
tensorboard
torch
torchdiffeq
tqdm
umnn
ninja
scikit-learn
pandas<2.0
h5py
