flax
Plumb through the prevent_cse kwarg for jax remat.
#1499

Merged

Plumb through the prevent_cse kwarg for jax remat. #1499

copybara-service merged 1 commit into google:main from levskaya:rematfix

google-cla added cla: yes

levskaya requested a review from

jheek 4 years ago

levskaya added pull ready

levskaya removed review request from

jheek 4 years ago

levskaya requested a review from

andsteing 4 years ago

levskaya requested a review from

avital 4 years ago

avital approved these changes on 2021-08-23

levskaya removed cla: yes

levskaya removed pull ready

levskaya added cla: yes

levskaya added pull ready

Plumb through the prevent_cse kwarg for jax remat.

7ed82a49

levskaya force pushed to 7ed82a49 4 years ago

copybara-service merged be522b51 into main 4 years ago

Reviewers

avital

andsteing

Assignees

No one assigned

Labels

cla: yes pull ready

Milestone

No milestone