DeepSpeed
feat(activation_checkpointing): add `non_reentrant_checkpoint` to support inputs require no grad
#4118
Merged
Merged