PR #4373 Add `flax.nnx.remat` docstring

Add `flax.nnx.remat` docstring #4373

copybara-service merged 1 commit into google:main from 8bitmp3:add-nnx-remat-docstring

8bitmp3 requested a review from

levskaya 1 year ago

8bitmp3 requested a review from

cgarciae 1 year ago

8bitmp3 requested a review from

IvyZX 1 year ago

8bitmp3 assigned

8bitmp3 1 year ago

IvyZX approved these changes on 2024-11-14

Add flax.nnx.remat docstring

8fbe99cf

8bitmp3 force pushed to 8fbe99cf 1 year ago

8bitmp3 added pull ready

copybara-service merged 856b6b58 into main 1 year ago

8bitmp3 deleted the add-nnx-remat-docstring branch 1 year ago

Reviewers

IvyZX

levskaya

cgarciae

Assignees

8bitmp3

Labels

pull ready

Milestone

No milestone