Skip to main content

Module layers

Module layers 

Source
Expand description

ESMFold2 layer implementations.

Modules§

atom_encoder
Atom encoder: per-atom feature encoder for ESMFold2.
confidence_head
Confidence head: predicts pLDDT, pAE, pDE, and distogram.
diffusion
AF3-style EDM diffusion module for all-atom coordinate generation.
folding_trunk
Folding trunk: 24-layer Pairformer transformer.
lm_encoder
LM encoder adapter: projects ESMC-6B hidden states → single representation.
pair_init
Pair representation initialization for the ESMFold2 FoldingTrunk.
pairformer
Pairformer block — core iterative refinement layer of the ESMFold2 FoldingTrunk.