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

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

6f93ad3
Select commit
Loading
Failed to load commit list.