scxpand-cuda 0.3.7.dev2

Pan-cancer detection of T-cell clonal expansion from single-cell RNA sequencing (CUDA-enabled)

scXpand: Pan-cancer Detection of T-cell Clonal Expansion

Detect T-cell clonal expansion from single-cell RNA sequencing data without paired TCR sequencing

Preprint • Documentation • Installation • Quick Start • Usage Guide • Citation

A framework for predicting T-cell clonal expansion from single-cell RNA sequencing data.

Manuscript in preparation - detailed methodology and benchmarks coming soon.

scXpand follows the scverse ecosystem <https://scverse.org/>_ standards for single-cell analysis tools.

View full documentation for comprehensive guides and API reference.

---

Features

• Multiple Model Architectures:
  • Autoencoder-based: Encoder-decoder with reconstruction and classification heads
  • MLP: Multi-layer perceptron
  • LightGBM: Gradient boosted decision trees
  • Linear Models: Logistic regression and support vector machines
• Scalable Processing: Handles millions of cells with memory-efficient data streaming from disk during training
• Automated Hyperparameter Optimization: Built-in Optuna integration for model tuning

---

Installation

Installing the Published Package: For regular use

scXpand is available in two variants to match your hardware:

NVIDIA GPU with CUDA Support

• Using pip:

  pip install --upgrade scxpand-cuda --extra-index-url https://download.pytorch.org/whl/cu128
  

• Using uv:

  uv pip install --upgrade scxpand-cuda --extra-index-url https://download.pytorch.org/whl/cu128 --index-strategy unsafe-best-match
  

CPU, Apple Silicon, or Non-CUDA GPUs

• Using pip:

  pip install --upgrade scxpand
  

• Using uv:

  uv pip install --upgrade scxpand
  

Local Development Setup: For contributing or working with the source code

If you want to contribute or work with the latest source code, follow these steps:

1. Clone the repository:

   git clone https://github.com/yizhak-lab-ccg/scXpand.git
   cd scXpand
   

2. Install uv (if not already installed):

   # Follow the installation guide: https://docs.astral.sh/uv/getting-started/installation/
   # Verify installation:
   uv --version
   

3. Install in development mode:

   uv pip install -e ".[dev]"
   

4. Install pre-commit hooks:

   pre-commit install
   

5. For PyTorch installation (optional):

   Using uv with automatic backend selection (Recommended):

   uv pip install torch torchvision torchaudio --torch-backend=auto
   

   This automatically detects your system's optimal PyTorch backend (CUDA if available, otherwise CPU/MPS).

See the full installation guide for detailed setup instructions.

---

Quick Start

import scxpand
# Make sure that "your_data.h5ad" includes only T cells for the results to be meaningful
# Ensure that "your_data.var_names" are provided as Ensembl IDs (as the pre-trained models were trained using this gene representation)
# Please refer to our documentation for more information

# List available pre-trained models
scxpand.list_pretrained_models()

# Run inference with automatic model download
results = scxpand.run_inference(
    model_name="pan_cancer_autoencoder",  # default model
    data_path="your_data.h5ad"
)

# Access predictions
predictions = results.predictions
if results.has_metrics:
    print(f"AUROC: {results.get_auroc():.3f}")

See our Tutorial Notebook for a complete example with data preprocessing, T-cell filtering, gene ID conversion, and model application using a real breast cancer dataset.

---

Documentation

Setup & Getting Started:

• Installation Guide - Setup for local development of scXpand
• User Guide - Quick start and comprehensive workflow guide
• Data Format - Input data requirements and specifications

Using Pre-trained Models:

• Model Inference - Run predictions on new data with pre-trained models

Training Your Own Models:

• Model Training - Train models with CLI and programmatic API
• Hyperparameter Optimization - Automated model tuning with Optuna

Understanding Results:

• Model Architectures - Detailed architecture descriptions and configurations
• Evaluation Metrics - Performance assessment and interpretation
• Output Format - Understanding model outputs and results

📖 Full Documentation - Complete guides, API reference, and interactive tutorials

---

License

This project is licensed under the MIT License – see the LICENSE file for details.

---

Citation

If you use scXpand in your research, please cite:

Shorer, O., Amit, R., and Yizhak, K. (2025). scXpand: Pan-cancer detection of T-cell clonal expansion from single-cell RNA sequencing without paired single-cell TCR sequencing. Preprint at bioRxiv, https://doi.org/10.1101/2025.09.14.676069.

BibTeX

@article{shorer2025scxpand,
  title={scXpand: Pan-cancer detection of T-cell clonal expansion from single-cell RNA sequencing without paired single-cell TCR sequencing},
  author={Shorer, Ofir and Amit, Ron and Yizhak, Keren},
  year={2025},
  journal={bioRxiv},
  doi={https://doi.org/10.1101/2025.09.14.676069}
}

---

This project was created in favor of the scientific community worldwide, with a special dedication to the cancer research community.

We hope you'll find this repository helpful, and we warmly welcome any requests or suggestions - please don't hesitate to reach out!