返回岗位列表

MiniMax

机器学习平台研发-k8s 方向

地点:北京 薪资:薪资未公开 日期:2025-03-25

岗位摘要

负责 MiniMax 机器学习平台的设计、开发和优化,保证模型数据处理、预训练、SFT、RL等模态各阶段任务的调度、拉起和稳定运行;负责云原生集群的控制器及调度器的研发和优化,优化大模型训练过程中的资源效率及成本

岗位职责

  • 负责 MiniMax 机器学习平台的设计、开发和优化,保证模型数据处理、预训练、SFT、RL等模态各阶段任务的调度、拉起和稳定运行
  • 负责云原生集群的控制器及调度器的研发和优化,优化大模型训练过程中的资源效率及成本
  • 负责Pytorch、Ray、Megatron等大模型训练框架以及各种分布式任务框架的支持和适配,解决不同框架下训练过程中出现的问题
  • 提供模型训练的指标上报、存储和观测能力

任职要求

  • 扎实的编程基础,精通至少一种高级编程语言,如Go, Python 等,具备良好的代码风格和工程习惯
  • 具有k8s集群的实操经验,有云原生大规模训练任务和推理服务的编排、在离线混部及资源调度经验者优先
  • 具有系统性能分析和优化的经验,具备独立分析解决问题的能力,有良好的沟通能力和团队协作精神
  • 了解 Pytorch/Tensorflow/JAX/PaddlePaddle/Mindspore等机器学习框架、GPU/NPU/ARM等最新异构计算系统与架构、RDMA高性能网络, 有相关系统研发经验者优先