Flashzoi RF 64k โ€” Trained from Scratch

Borzoi architecture trained from random initialization on mouse ESC-derived CUT&Tag H3K27me3 and H3K4me3 tracks.

  • Receptive field: 64k
  • Architecture: Borzoi (from scratch, model name "flashzoi")
  • Resolution: 32 bp
  • Training data: Normalized CUT&Tag from mouse ESC-derived embryoid bodies
  • Paper: Casimov, Lifshitz et al., "Reverse engineering the genomic encodings of the pluripotent epigenome"

Usage

torchrun --nproc_per_node=8 infer_borzoi_pytorch.py \
    --config config.yaml \
    --checkpoint model.safetensors \
    --genome_fasta mm10.fa \
    --output_dir predictions/

See the paper companion repository for full code and configs.

Downloads last month
14
Safetensors
Model size
0.2B params
Tensor type
F32
ยท
Inference Providers NEW
This model isn't deployed by any Inference Provider. ๐Ÿ™‹ Ask for provider support

Collection including tanaylab/sns-paper-flashzoi-rf64k