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.