校招- AI Infra工程师（训练优化方向）(J18090)当面告知
零跑汽车有限公司
2026-11-19 17:27:42
校招博士27届硕士
岗位详情
地区：
杭州市
专业要求：
人工智能自动化软件工程计算机类电子信息类
岗位职责：
1、负责大模型分布式训练系统的性能优化与工程化建设，为算法团队提供高吞吐、高可用的训练能力，支撑千卡规模大模型训练的高效落地；
2、参与分布式训练框架的封装与优化，基于 PyTorch/Megatron-LM/DeepSpeed/FSDP 等框架，深入应用数据并行（DP）、张量并行（TP）、流水线并行（PP）、ZeRO 等并行策略，优化大模型训练的扩展性与资源利用率；
3、负责大模型训练中的显存管理与通信优化，深入理解并优化 NCCL 通信机制、梯度同步、混合精度训练等关键环节，降低通信开销与显存峰值；
4、开展 GPU 高性能计算优化，包括算子融合、访存优化、低比特训练等前沿技术的探索与落地，持续提升训练吞吐与稳定性；
5、开发并维护训练性能诊断工具（Profiling），协助算法团队分析千卡训练中的通信瓶颈、算子耗时与显存占用，提供端到端的吞吐量、MFU、显存峰值等核心指标的量化分析与优化建议；
6、探索分布式训练前沿技术（如长序列训练、序列并行、异构训练等），沉淀训练最佳实践与代码模板，与 AI Infra 团队协作落地高性能训练方案。
招聘要求：
1、硕士及以上学历，计算机类、电子信息类、软件工程、人工智能、自动化等相关专业；
2、扎实的计算机基础，掌握数据结构与算法、操作系统、计算机网络等核心知识；熟练掌握 Python，熟悉异步/并发编程，具备良好的代码规范，了解 C++、CUDA 编程；
3、熟悉 Linux 环境与容器化基本功，掌握 Docker 基本使用、Shell 脚本及常见调试排障方法；理解异步任务队列、限流/重试/熔断等分布式系统基本概念；
4、对训练优化方向有课程项目、竞赛或实习经历：熟悉分布式训练并行策略（DP/TP/PP/ZeRO），了解大模型显存管理与通信机制（NCCL），有 Megatron/DeepSpeed/FSDP 使用经验，了解混合精度、梯度累积、序列并行等技术；
5、加分项（任一即可）：1. 有 CUDA/Triton 算子编写与性能调优经验；2. 参与过百亿/千亿参数大模型从 0 到 1 的分布式训练落地；3. 熟悉 NCCL 源码或有过集合通信优化经验；4. 有自动驾驶（感知/规划/决策）大规模训练工程化经验；5. 有开源贡献/技术博客/完整工程作品集（GitHub/Gitee）；
6、具备良好的沟通协作与文档能力，主动性强，愿意跨团队合作落地 AI 能力。
公司介绍
零
零跑汽车有限公司
零跑汽车有限公司，成立于2017-01-12，位于浙江，法定代表人：朱江明，登记状态：存续，注册资本470800万人民币，公司类型：民营/私企
公司信息:公司名称:零跑汽车有限公司公司类型:机械制造/机电/重工公司性质:民营企业公司地点:浙江省金华市公司规模:5000以上涉及领域:--公司网址:--