Commit History

Author SHA1 Message Date
  Masahiro Tanaka 2c51aba0b7 Add custom reshaping for universal checkpoint (#5390) 6 months ago
  Masahiro Tanaka c56a4b9e0d Improve universal checkpoint (#5289) 6 months ago
  Masahiro Tanaka b112c99ea8 Fix loading a universal checkpoint (#5263) 7 months ago
  Sam Ade Jacobs 8998707a2f Universal Checkpoint for Sequence Parallelism (#4752) 10 months ago
  Moshe Island ce5e56a82e universal-ckp: support megatron-deepspeed llama model (#4666) 11 months ago
  Moshe Island 8ad187d84f Universal ckp fixes (#4588) 11 months ago
  Michael Wyatt b361c72761 Update DeepSpeed copyright license to Apache 2.0 (#3111) 1 year ago
  Jeff Rasley 91d63e0228 update formatter version and style settings (#3098) 1 year ago
  Olatunji Ruwase 799120e7e4 Universal checkpoint for zero stage 1 (#2284) 2 years ago
  Olatunji Ruwase 53182531ed Refactor universal checkpointing and tensor fragments (#2253) 2 years ago