返回岗位列表

壁仞科技

分布式训练工程师

地点:上海 薪资:30-60K 日期:2026-03-03

岗位摘要

基于公司生态,针对NLP/CV/音频/多模态/推广搜等场景,构建大规模分布式机器学习系统;解决业务交付流程中单机多卡、多机多卡的精度与性能问题;研究行业领先的超大规模分布式策略,攻克大模型落地中的分布式工程挑战

岗位职责

  • 基于公司生态,针对NLP/CV/音频/多模态/推广搜等场景,构建大规模分布式机器学习系统
  • 解决业务交付流程中单机多卡、多机多卡的精度与性能问题
  • 研究行业领先的超大规模分布式策略,攻克大模型落地中的分布式工程挑战

任职要求

  • 计算机、电子、数学及相关专业,2年以上相关工作经验
  • 熟练掌握C++/Python,有CUDA编程经验优先
  • 熟悉TensorFlow/PyTorch/Paddle等主流及国产机器学习框架,有框架开发经验优先
  • 熟悉CUDA生态,具备NCCL、RDMA开发经验优先
  • 了解常见大规模分布式训练优化策略,熟悉Megatron/DeepSpeed/Accelerate/Galvatron者优先

福利待遇

  • 年薪30-60K×15薪,极具竞争力
  • 参与前沿大模型分布式系统建设,技术成长快
  • 上海闵行区核心科技园区办公,环境优越

工作地址

上海闵行区壁仞科技