Fix megatron grad norm explosion#648

Open

Kovbo wants to merge 5 commits intomainfrom

fix/megatron-gradient-explosion

Commits on Apr 9, 2026

fix drad norm explosion
Kovbo
committed

Commits on Apr 10, 2026

fix types
Kovbo
committed

Commits on Apr 13, 2026

Use aot_eager for shared-prefix block masks
FurtherAI
committed

Commits on Apr 14, 2026