torch>=1.11.0
torchvision>=0.11.0
lightning-bolts
matplotlib
numpy
scipy
scikit-learn
scprep
scanpy
torchdyn
pot
torchdiffeq
absl-py
clean-fid

[forest-flow]
xgboost
scikit-learn
ForestDiffusion
