Skip to content

fix(transform): cast updates to grad dtype for mixed-precision support#1687

Open
nileshpatil6 wants to merge 1 commit into
google-deepmind:mainfrom
nileshpatil6:fix/optimizer-dtype-tracks-grads
Open

fix(transform): cast updates to grad dtype for mixed-precision support#1687
nileshpatil6 wants to merge 1 commit into
google-deepmind:mainfrom
nileshpatil6:fix/optimizer-dtype-tracks-grads

Commits

Commits on May 30, 2026