requirements.txt

wandb
appnope==0.1.2
argon2-cffi==21.3.0
argon2-cffi-bindings==21.2.0
asttokens==2.0.5
attrs==21.4.0
backcall==0.2.0
black==21.12b0
bleach==4.1.0
boltons==21.0.0
cffi==1.15.0
click==8.0.3
cycler==0.11.0
debugpy==1.5.1
decorator==5.1.1
defusedxml==0.7.1
entrypoints==0.3
executing==0.8.2
fonttools==4.29.0
jedi==0.18.1
Jinja2==3.0.3
jsonschema==4.4.0
kiwisolver==1.3.2
MarkupSafe==2.0.1
mistune==0.8.4
mypy-extensions==0.4.3
nbclient==0.5.10
nbconvert==6.4.0
nbformat==5.1.3
nest-asyncio==1.5.4
notebook==6.4.8
packaging==21.3
pandocfilters==1.5.0
parso==0.8.3
pathspec==0.9.0
pexpect==4.8.0
pickleshare==0.7.5
Pillow==9.0.0
platformdirs==2.4.1
plotly==5.5.0
prometheus-client==0.13.0
prompt-toolkit==3.0.24
ptyprocess==0.7.0
pure-eval==0.2.2
pycparser==2.21
Pygments==2.11.2
pyparsing==3.0.7
pyrsistent==0.18.1
python-dateutil==2.8.2
pytz==2021.3
pyzmq==22.3.0
qtconsole==5.2.2
QtPy==2.0.0
Send2Trash==1.8.0
stack-data==0.1.4
tenacity==8.0.1
terminado==0.12.1
testpath==0.5.0
tomli==1.2.3
torch==1.13.1
torchsde==0.2.5
torchvision
tornado==6.1
traitlets==5.1.1
trampoline==0.1.2
typing-extensions==4.0.1
wcwidth==0.2.5
webencodings==0.5.1
widgetsnbextension==3.5.2
ipdb~=0.13.9
torchdiffeq
geoopt~=0.4.0
hydra-core==1.2.0
hydra-colorlog==1.2.0
hydra-optuna-sweeper==1.2.0
hydra-submitit-launcher
torchdyn==1.0.6
scipy
scikit-learn
pot
scprep
scanpy
lightning-bolts
timm
seaborn>=0.12.2
pandas
pytorch-lightning==1.8.3
ipdb
einops
git+https://github.com/joeybose/geomstats.git@master