DeepSpeed

52c87a7c - Merge branch 'clean-opt' into lekurile/ds_chat_mlp_debug