feat(activation_checkpointing): add non_reentrant_checkpoint to support inputs require no grad - #4118
Merged
tjruwase merged 27 commits intoAug 29, 2023
Merged
non_reentrant_checkpoint to support inputs require no grad - #4118