Skip to main content

Module folding_trunk

ferritin_plms::esmfold2::layers

Module folding_trunk 

Source
Expand description

Folding trunk: 24-layer Pairformer transformer.

Weight layout:

folding_trunk.pair_init.*            — pair representation initializer
folding_trunk.blocks.{0..23}.*       — Pairformer layers
folding_trunk.norm.*                 — final single-repr LayerNorm (TODO)

Structs§

FoldingTrunk
24-layer Pairformer trunk that jointly refines single and pair representations.