PR #2457 Add static_argnums to nn.checkpoint

Add static_argnums to nn.checkpoint #2457

copybara-service merged 2 commits into google:main from cgarciae:fix-checkpoint

cgarciae changed the title ~~Forward static_argnums to remat~~ Add static_argnums to nn.checkpoint 3 years ago

cgarciae requested a review from

IvyZX 3 years ago

cgarciae assigned

cgarciae 3 years ago

cgarciae added Priority: P1 - soon

add static_argnums to nn.checkpoint

9522bc0b

cgarciae force pushed to 9522bc0b 3 years ago

IvyZX approved these changes on 2022-09-13

cgarciae added pull ready

cgarciae removed pull ready

correct argnum indexes for transforms.checkpoint

f384b331

IvyZX approved these changes on 2022-09-13

cgarciae requested a review from

IvyZX 3 years ago

cgarciae added pull ready

copybara-service merged 59280a36 into main 3 years ago

Reviewers

IvyZX

Assignees

cgarciae

Labels

Priority: P1 - soon pull ready

Milestone

No milestone