flax
Fix dtype handling for Adam and LAMB optimizers in 64bit mode.
#965

Merged

Fix dtype handling for Adam and LAMB optimizers in 64bit mode. #965

copybara-service merged 2 commits into google:master from levskaya:optimizer_fixes

google-cla added cla: yes

levskaya added pull ready

Fix dtype handling for Adam and LAMB optimizers in 64bit mode.

dc841b04

levskaya force pushed to dc841b04 4 years ago

levskaya requested a review from

andsteing 4 years ago

marcvanzee assigned

levskaya 4 years ago

marcvanzee added Priority: P1 - soon

Merge branch 'master' into optimizer_fixes

f9038b17

levskaya requested a review from

avital 4 years ago

avital approved these changes on 2021-02-02

copybara-service merged 61580b99 into master 4 years ago

Reviewers

avital

andsteing

Assignees

levskaya

Labels

cla: yes Priority: P1 - soon pull ready

Milestone

No milestone