Installation

pip install prism-de

The import name is prism. PRISM needs Python >= 3.10 and PyTorch >= 2.2.

GPU

Fits run on the GPU by default (device="cuda"). Install the PyTorch build that matches your CUDA driver first, then PRISM. For example, for CUDA 12.4:

pip install torch --index-url https://download.pytorch.org/whl/cu124
pip install prism-de

See pytorch.org for other platforms. Without a GPU, pass device="cpu" to PRISMConfig and PrismData.from_anndata.

Optional extras

pip install "prism-de[wandb]"   # Weights & Biases logging

Check the install

python -c "import prism, torch; print(prism.__version__, torch.__version__, torch.cuda.is_available())"

The paper's results were produced with Python 3.10 and PyTorch 2.5.1 (CUDA 12.4).