智元|训练框架研发实习生

智元 · 上海 · 更新时间:2026-09-01

岗位职责

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. 具备相关领域的研发和交付经验,深刻理解训练研发框架的定位和意义。

智元招聘实习生:训练框架研发实习生岗位来自智元招聘官网,具体内容以智元招聘官网为准。