Module output_heads 

ESM3 output projection heads (OutputHeads).

Projects the transformer hidden state to per-track logit distributions.

Structs

ESM3Output

Output of an ESM3 forward pass.

OutputHeads

RegressionHead

Linear(d_in) → GELU → LayerNorm → Linear(d_out).