简化版的训练框架，全部由C++实现，包含了autograd机制、kernel层关键算子实现，测例中包含单点验证以及端到端训练GPT2模型流程，支持CPU和CUDA平台。
BAI-123-GUO/TinyInfiniTrain
Folders and files
| Name | Name | Last commit date | ||
|---|---|---|---|---|
Repository files navigation
Releases
No releases published
Languages
- C++ 68.8%
- Cuda 28.3%
- Python 1.4%
- Other 1.5%