evo2 SAE: inference engine + steering, CI lane, tests, Dockerfile#1622

Open

polinabinder1 wants to merge 11 commits into

mainfrom

pbinder/evo2-sae-serve

Commits on Jun 23, 2026

evo2 SAE: inference engine + steering, CI lane, and tests (on migrated layout)

polinabinder1
and
claude
committed
test(evo2-sae): cover non-finite + contract edge cases in steering sanitize

polinabinder1
and
claude
committed
evo2-sae: guard negative top_k in sanitize + cover empty-sequence encode_batch

polinabinder1
and
claude
committed
evo2-sae: add runnable Dockerfile (thin wrapper over .ci_build.sh)

polinabinder1
and
claude
committed
fix(evo2-sae): TopK-only — steer via the SAE's own encode, drop ReLU

polinabinder1
and
claude
committed
fix(evo2-sae): conftest loads the 1B via bionemo.common (was bionemo.core)

polinabinder1
and
claude
committed
test(evo2-sae): register the 'slow' pytest marker in the recipe pyproject

polinabinder1
and
claude
committed
build(evo2-sae): cache the megatron build separately from SAE code (Docker layers)

polinabinder1
and
claude
committed
feat(evo2-sae): fail loudly on an SAE/model dim mismatch at load

polinabinder1
and
claude
committed
evo2-sae: length-bucket encode_batch to cut padding waste

polinabinder1
and
claude
committed
evo2-sae: collapse to a single inference engine

polinabinder1
and
claude
committed