简化版的训练框架,全部由C++实现,包含了autograd机制、kernel层关键算子实现,测例中包含单点验证以及端到端训练GPT2模型流程,支持CPU和CUDA平台。
forked from InfiniTensor/TinyInfiniTrain
-
Couldn't load subscription status.
- Fork 0
2025夏季训练营训练方向项目
License
Couldn't load subscription status.
Aoi979/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.2%
- Cuda 28.8%
- Python 1.5%
- Other 1.5%