numpy
torch
pot
scikit-learn
tqdm
geoopt  # For hyperbolic Gradient FLows