Pip install the package locally to use the models. The requisite package list is not set yet, so you will have to manuall install the following packages for now.
- Pytorch
- Matplotlib
- Torchdiffeq (consider this version instead which has proximal backward solvers)
- Tqdm
pip install -e .