返回岗位列表

壁仞科技

分布式训练工程师

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

岗位摘要

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

岗位职责

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

任职要求

  • 计算机、电子、数学等相关专业硕士,2年以上相关经验
  • 精通C++/Python,有CUDA开发经验优先
  • 熟悉TensorFlow/PyTorch/Paddle等框架,有框架源码开发经验优先
  • 熟悉CUDA生态,具备NCCL、RDMA开发经验优先
  • 了解Megatron、DeepSpeed、Accelerate、Galvatron等分布式训练优化策略

福利待遇

  • 薪资25-50K,硕士学历优先
  • 杭州滨江区海威天地T3现代化办公环境
  • 参与前沿大模型分布式系统建设,技术成长空间大

工作地址

杭州滨江区海威天地T3-28