Add automatic mixed precision training for ATME#2

benbennett wants to merge 1 commit intomainbenbennett/SIMPLE:mainfrom

codex/find-ways-to-speed-up-trainingbenbennett/SIMPLE:codex/find-ways-to-speed-up-trainingCopy head branch name to clipboard