机器学习平台研发工程师-K8s方向
岗位摘要
负责MiniMax机器学习平台的设计、开发和优化,保证模型数据处理、预训练、SFT、RL等模态各阶段任务的调度、拉起和稳定运行;负责云原生集群的控制器及调度器的研发和优化,优化大模型训练过程中的资源效率及成本
登录后保留你的岗位线索
邮箱登录后可继续查看完整职责、任职要求和原岗位入口,并使用收藏与浏览记录沉淀适合自己的机会。
岗位职责
- 负责MiniMax机器学习平台的设计、开发和优化,保证模型数据处理、预训练、SFT、RL等模态各阶段任务的调度、拉起和稳定运行
- 负责云原生集群的控制器及调度器的研发和优化,优化大模型训练过程中的资源效率及成本
- 负责Pytorch、Ray、Megatron等大模型训练框架以及各种分布式任务框架的支持和适配,解决不同框架下训练过程中出现的问题
- 提供模型训练的指标上报、存储和观测能力
任职要求
- 扎实的编程基础,精通至少一种高级编程语言,如Go、Python等,具备良好的代码风格和工程习惯
- 具有K8s集群的实操经验,有云原生大规模训练任务和推理服务的编排、在离线混部及资源调度经验者优先
- 具有系统性能分析和优化的经验,具备独立分析解决问题的能力,有良好的沟通能力和团队协作精神
- 了解Pytorch/Tensorflow/JAX/PaddlePaddle/Mindspore等机器学习框架、GPU/NPU/ARM等异构计算系统与架构、RDMA高性能网络,有相关系统研发经验者优先
福利待遇
- 具有竞争力的薪酬待遇:40-60K·16薪
- 加入MiniMax,与顶尖AI团队共同成长
- 参与大模型训练平台核心系统研发,技术成长空间大
工作地址
北京海淀区蓟门壹号
登录后查看完整岗位详情
使用邮箱验证码登录后,可查看完整职责、任职要求、原岗位入口,并继续使用收藏和浏览记录。列表摘要保持公开,完整详情用于保护数据质量和降低恶意抓取。