flax
d7172e9d - Merge pull request #1560 from jheek:add-checkpoint-policy-api

Commit

4 years ago

Merge pull request #1560 from jheek:add-checkpoint-policy-api PiperOrigin-RevId: 398808974

References

#1560 - Forward experimental checkpoint policy option in lift.remat

Author

a-googler

a-googler

Committer

a-googler

a-googler

Parents

Loading