Skip to content

add sharded checkpoint loading for AutoTP path to reduce the peak mem… - #3102

Merged
tjruwase merged 10 commits into
deepspeedai:masterfrom
sywangyi:autoTP_memory_opt
May 4, 2023
Merged

tjruwase merged 10 commits into
deepspeedai:masterfrom
sywangyi:autoTP_memory_opt

Commits

Commits on Mar 29, 2023

Commits on Mar 30, 2023

Commits on Apr 18, 2023

Commits on Apr 21, 2023

Commits on Apr 22, 2023

Commits on Apr 26, 2023

Commits on May 3, 2023

Commits on May 4, 2023