DeepSpeed
Activation checkpointing for non-tensor arguments and return values
#741
Merged
Merged