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

Merge branch 'master' into autoTP_memory_opt

227ee47
Select commit
Loading
Failed to load commit list.

Workflow runs completed with no jobs