requirements.txt

torch>=2.0
numpy
scipy
scikit-learn
pandas
pyarrow
scanpy
anndata
matplotlib

# gears is only needed for the head-to-head baseline and lives in a separate env
# (see scripts/setup_gears_env.sh) because it pins older torch/pyg versions.