helical 1.4.6

Helical Python SDK

What is Helical ?

Helical provides a framework for state-of-the-art pre-trained bio foundation models on genomics and transcriptomics modalities.

Helical simplifies the entire application lifecycle when building with bio foundation models. You will be able to:

• Leverage the latest bio foundation models through our easy-to-use python package
• Run example notebooks on key downstream tasks from examples

We will update this repo on a regular basis with new models, benchmarks, modalities and functions - so stay tuned. Let’s build the most exciting AI-for-Bio community together!

         

What's new?

New Larger Geneformer Models

We have integrated the new Geneformer models which are larger and have been trained on more data. Find out which models have been integrated into the Geneformer suite in the model card. Check out the our notebook on drug perturbation prediction using different Geneformer scalings here.

TranscriptFormer

We have integrated TranscriptFormer into our helical package and have made a model card for it in our Transcriptformer model folder. If you would like to test the model, take a look at our example notebook!

🧬 Introducing Helix-mRNA-v0: Unlocking new frontiers & use cases in mRNA therapy 🧬

We’re thrilled to announce the release of our first-ever mRNA Bio Foundation Model, designed to:

1. Be Efficient, handling long sequence lengths effortlessly
2. Balance Diversity & Specificity, leveraging a 2-step pre-training approach
3. Deliver High-Resolution, using single nucleotides as a resolution

Check out our blog post to learn more about our approach and read the model card to get started.

Installation

We recommend installing Helical within a conda environment with the commands below (run them in your terminal) - this step is optional:

conda create --name helical-package python=3.11.13
conda activate helical-package

To install the latest pip release of our Helical package, you can run the command below:

pip install helical

To install the latest Helical package, you can run the command below:

pip install --upgrade git+https://github.com/helicalAI/helical.git

Alternatively, clone the repo and install it:

git clone https://github.com/helicalAI/helical.git
pip install .

[Optional] To install mamba-ssm and causal-conv1d use the command below:

pip install helical[mamba-ssm]

or in case you're installing from the Helical repo cloned locally:

pip install .[mamba-ssm]

Notes on the installation:

• Make sure your machine has GPU(s) and Cuda installed. Currently this is a requirement for the packages mamba-ssm and causal-conv1d.
• The package causal_conv1d requires torch to be installed already. First installing helical separately (without [mamba-ssm]) will install torch for you. A second installation (with [mamba-ssm]), installs the packages correctly.
• If you have problems installing mamba-ssm, you can install the package via the provided .whl files on their release page here. Choose the package according to your cuda, torch and python version:

pip install https://github.com/state-spaces/mamba/releases/download/v2.2.4/mamba_ssm-2.2.4+cu12torch2.3cxx11abiFALSE-cp311-cp311-linux_x86_64.whl

• Now continue with pip install .[mamba-ssm] to also install the remaining causal-conv1d.

Singularity (Optional)

If you desire to run your code in a singularity file, you can use the singularity.def file and build an apptainer with it:

apptainer build --sandbox singularity/helical singularity.def

and then shell into the sandbox container (use the --nv flag if you have a GPU available):

apptainer shell --nv --fakeroot singularity/helical/

RNA models:

• Helix-mRNA
• Mamba2-mRNA
• Geneformer
• scGPT
• Universal Cell Embedding (UCE)
• TranscriptFormer

DNA models:

• HyenaDNA
• Caduceus
• Evo 2

Demo & Use Cases

To run examples, be sure to have installed the Helical package (see Installation) and that it is up-to-date.

You can look directly into the example folder above and download the script of your choice, look into our documentation for step-by-step guides or directly clone the repository using:

git clone https://github.com/helicalAI/helical.git

Within the examples/notebooks folder, open the notebook of your choice. We recommend starting with Quick-Start-Tutorial.ipynb

Current Examples:

Example	Description	Colab
Quick-Start-Tutorial.ipynb	A tutorial to quickly get used to the helical package and environment.
Helix-mRNA.ipynb	An example of how to use the Helix-mRNA model.
Geneformer-vs-TranscriptFormer.ipynb	Zero-Shot Reference Mapping with Geneformer & TranscriptFormer and compare the outcomes.
Hyena-DNA-Inference.ipynb	An example how to do probing with HyenaDNA by training a neural network on 18 downstream classification tasks.
Cell-Type-Annotation.ipynb	An example how to do probing with scGPT by training a neural network to predict cell type annotations.
Cell-Type-Classification-Fine-Tuning.ipynb	An example how to fine-tune different models on classification tasks.
HyenaDNA-Fine-Tuning.ipynb	An example of how to fine-tune the HyenaDNA model on downstream benchmarks.
Cell-Gene-Cls-embedding-generation.ipynb	A notebook explaining the different embedding modes of single cell RNA models.
Geneformer-Series-Comparison.ipynb	A zero shot comparison between Geneformer model scaling on drug perturbation prediction

Stuck somewhere ? Other ideas ?

We are eager to help you and interact with you:

• Join our Slack channel where you can discuss applications of bio foundation models.
• You can also open Github issues here.

Why should I use Helical & what to expect in the future?

If you are (or plan to) working with bio foundation models s.a. Geneformer or UCE on RNA and DNA data, Helical will be your best buddy! We provide and improve on:

• Up-to-date model library
• A unified API for all models
• User-facing abstractions tailored to computational biologists, researchers & AI developers
• Innovative use case and application examples and ideas
• Efficient data processing & code-base

We will continuously upload the latest model, publish benchmarks and make our code more efficient.

Contributing

We welcome all kinds of contributions, including code, documentation, bug reports, and feature suggestions. Please read our Contributing Guidelines to help us keep the project organized and collaborative.

Acknowledgements

A lot of our models have been published by talend authors developing these exciting technologies. We sincerely thank the authors of the following open-source projects:

• scGPT
• Geneformer
• UCE
• TranscriptFormer
• HyenaDNA
• anndata
• scanpy
• transformers
• scikit-learn
• GenePT
• Caduceus
• Evo2
• torch
• torchvision

Licenses

You can find the Licenses for each model implementation in the model repositories:

• Helix-mRNA
• Mamba2-mRNA
• scGPT
• Geneformer
• UCE
• TranscriptFormer
• HyenaDNA
• Evo2

Citation

Please use this BibTeX to cite this repository in your publications:

@software{allard_2024_13135902,
  author       = {Helical Team},
  title        = {helicalAI/helical: v1.1.0},
  month        = nov,
  year         = 2024,
  publisher    = {Zenodo},
  version      = {1.1.0},
  doi          = {10.5281/zenodo.13135902},
  url          = {https://doi.org/10.5281/zenodo.13135902}
}