DeepSpeed

dd8df20f - zero3 checkpoint frozen params (#3205)

zero3 checkpoint frozen params (#3205) * zero3 checkpoint frozen params * Remove debug prints * Move to cpu * WIP * WIP * WIP * Cleanup * Cleanup * Extend unit test for frozen params * API fix