Skip to content

[JAX] Add checkpoint_name for the recompute granularity control#542

Merged
denera merged 1 commit intoNVIDIA:mainfrom zlsh80826:rewang/add_checkpoint_nameDec 4, 2023
+20

Commits

Commits on Dec 3, 2023