DeepSpeed

01b6e27e - Activation checkpointing bugfix and unit tests (#420)

Activation checkpointing bugfix and unit tests (#420) * Activation checkpointing bugfix and unit tests. * Activation checkpointing bugfix and unit tests. Co-authored-by: Jeff Rasley <jerasley@microsoft.com>