Skip to main content

Module atom_encoder

ferritin_plms::esmfold2::layers

Module atom_encoder 

Source
Expand description

Atom encoder: per-atom feature encoder for ESMFold2.

Weight layout:

inputs.atom_encoder.blocks.{0..2}.*   — 3 atom transformer blocks

Structs§

AtomEncoder
3-block atom-level transformer encoder.