
智元|训练框架研发实习生
岗位职责
1. 负责大模型分布式训练框架的设计和研发。
2. 负责异步数据加载模块的设计和研发。
3. 负责容器化开发环境的构建与维护。
4. 与算法工程师密切配合,提高训练效率,降低研发成本,提升交付能力。
任职要求
1. 具备坚实的 Python/C++ 编程基础。
2. 具备良好的代码规范和强烈的工程意识。
3. 至少熟悉 OpenMMLab / PyTorch Lightning / Hugging Face Transformers / Detectron2 / FastAI 其中一种训练研发框架。
4. 至少熟悉 Megatron-Core 或 Deepspeed 其中一种分布式训练框架。
5. 熟练掌握性能优化、量化、蒸馏等相关工具的使用。
6. 熟练掌握 Docker 等容器化工具的使用。
7. 具备相关领域的研发和交付经验,深刻理解训练研发框架的定位和意义。
智元招聘实习生:训练框架研发实习生岗位来自智元招聘官网,具体内容以智元招聘官网为准。
