DeepSpeed

ec8b1cb0 - Activation checkpointing for non-tensor arguments and return values (#741)

Activation checkpointing for non-tensor arguments and return values (#741) * Activation checkpoint support for non tensor input/output * Format fixes * Address PR comments; Add ordering edge case tests