[DRAFT] DO NOT MERGE Esm2 minifold#1541

Open

jomitchellnv wants to merge 20 commits intomainfrom

Commits on Mar 31, 2026

Commits on Apr 1, 2026

Commits on Apr 6, 2026

enables layer-wise and component-wise low precision targeting
jomitchellnv
committed
sets layer components low precision to false except ffn
jomitchellnv
committed
removes torch.autocast and uses fsdp2 for autocasting to bf16
jomitchellnv
committed

Commits on Apr 7, 2026