PR #1479 Updates `examples/lm1b` to use Optax

Updates `examples/lm1b` to use Optax #1479

copybara-service merged 6 commits into google:main from jaehlee:main

update lm1b example to use Optax.

c086d0eb

google-cla added cla: yes

Update instructions for running lm1b example

27e640ea

andsteing requested a review from

andsteing 4 years ago

andsteing assigned

andsteing 4 years ago

andsteing requested changes on 2021-08-06

match examples/wmt changes

97cd6657

fix args for `create_learning_rate_schedule`

36a0bf59

update results

2097ffeb

andsteing requested changes on 2021-08-12

change base learning_rate to preserve schedule

da9d8ef9

andsteing approved these changes on 2021-08-13

andsteing added pull ready

copybara-service merged 9bc7e2a9 into main 4 years ago

Reviewers

andsteing

Assignees

andsteing

Labels

cla: yes pull ready

Milestone

No milestone