返回岗位列表

壁仞科技

分布式训练工程师

地点:杭州 薪资:30-50K 日期:2026-03-03

岗位摘要

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

岗位职责

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

任职要求

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

福利待遇

  • 15薪,年薪30-50K区间,具备竞争力
  • 杭州滨江核心办公,交通便捷
  • 参与前沿大模型分布式系统建设,技术成长空间大

工作地址

杭州滨江区海威天地T328F