← all repositories

lucidrains/enformer-pytorch

A PyTorch implementation of DeepMind's Enformer attention network for predicting gene expression from DNA sequences.

566 stars Python Domain AppsML Frameworks
enformer-pytorch
Velocity · 7d
+0.3
★ / day
Trend
steady
star history

This repository provides a PyTorch port of DeepMind’s Enformer model, an attention-based neural network architecture designed for genomic analysis. The model processes DNA sequences and predicts gene expression levels across multiple organisms (human and mouse). It supports fine-tuning pretrained weights for downstream tasks such as chromatin accessibility prediction, and can extract embeddings for transfer learning applications.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.