Ivyea Jobs 6655 roles · 90 companies · 刚刚
智元机器人 · 具身智能

训练框架研发实习生

上海 实习 AI Infra / 训练系统

职位描述

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. 具备相关领域的研发和交付经验,深刻理解训练研发框架的定位和意义。
智元机器人去官方页面投递 →