numpy
einops
omegaconf
torchdiffeq>=0.2.3