feat: Enable amp with autocast (fix poor bf16 convergence on GRPO#26
Merged
SahilJain314 merged 2 commits intomainfrom Mar 21, 2025
Merged
feat: Enable amp with autocast (fix poor bf16 convergence on GRPO#26SahilJain314 merged 2 commits intomainfrom
SahilJain314 merged 2 commits intomainfrom