Support mixed precision training#219

Open

perctrix wants to merge 7 commits intoProjectNeura:mainfrom

perctrix:support-amp

Commits on Feb 19, 2026

Support mixed precision training (#179)
perctrix
committed

Commits on Feb 23, 2026

Merge branch 'main' into support-amp
perctrix
committed

Commits on Mar 1, 2026

Fix AMP scheduler drift and GradScaler device mismatch (#219)
perctrix
committed

Commits on Mar 6, 2026

Use `binary_cross_entropy_with_logits` for autocast safety (#179)
perctrix
committed

Commits on Mar 7, 2026

Guard EMA update on successful AMP optimizer step (#179)
perctrix
committed

Commits on Mar 16, 2026

Honor `amp=False` when resuming from a mixed-precision run (#219)
perctrix
committed

Commits on Mar 19, 2026

Cast soft-dice inputs to float32 for AMP compatibility (#219)
perctrix
committed