DeepSpeed

485a365f - ZeRO-Offload passing model tests (#374)

ZeRO-Offload passing model tests (#374) * Add ZeRO-Offload model tests Restrict optimizer update+copy to DeepSpeedCPUAdam * Format fixes * Increate bucket size scaler