Skip to content

A comparison of pretraining framework for LLM

Notifications You must be signed in to change notification settings

xdxd1234-bit/LLMTrainer

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

12 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

LLMTrainer

对比 LLM 的多种预训练方案的性能,方案包括:

  • Torch DDP
  • FSDP
  • DeepSpeed
  • FlashAttention V2
  • ColossalAI
  • Sophia Optimizer
  • Lightning Fabric

详细介绍见 https://zhuanlan.zhihu.com/p/648094197?

About

A comparison of pretraining framework for LLM

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 87.6%
  • Shell 12.4%