numpy
svgpathtools
matplotlib
tqdm
einops
ema-pytorch
torch
hydra-core
wandb
scikit-learn
geomstats
seaborn
torchdiffeq
pyvista[all]
healpy
e3nn
moviepy
