简化版的训练框架，全部由C++实现，包含了autograd机制、kernel层关键算子实现，测例中包含单点验证以及端到端训练GPT2模型流程，支持CPU和CUDA平台。
forked from InfiniTensor/TinyInfiniTrain
-
Notifications
You must be signed in to change notification settings - Fork 0
2025夏季训练营训练方向项目
License
awa-yw/TinyInfiniTrain
Folders and files
| Name | Name | Last commit message | Last commit date | |
|---|---|---|---|---|
Repository files navigation
About
2025夏季训练营训练方向项目
Resources
License
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published
Languages
- C++ 68.6%
- Cuda 28.4%
- Python 1.5%
- Other 1.5%