flax
e59b14e9 - Modify Flax checkpointing in preparation for cl/650338576.

Commit

1 year ago

Modify Flax checkpointing in preparation for cl/650338576. PiperOrigin-RevId: 651196174

References

Author

cpgaffney1

Committer

a-googler

Parents