校招- AI Infra工程师(工具链开发方向)(J18092)当面告知
零跑汽车有限公司
2026-11-19 17:27:42
校招博士27届硕士
岗位详情
地区:
杭州市
专业要求:
人工智能自动化软件工程计算机类电子信息类
岗位职责:
1、负责大模型训练与推理工具链的研发与迭代,为算法团队提供易用的 SDK 与工程化组件,降低模型开发、训练与部署的门槛;
2、参与分布式训练框架的封装与优化,基于 PyTorch/Megatron-LM/DeepSpeed等框架,提供易用的分布式启动、断点续训、容错恢复机制;
3、开发并维护训练性能诊断工具,协助算法团队分析千卡训练中的通信瓶颈、算子耗时与显存占用,支持预训练、SFT、RLHF 等场景的标准化模板;
4、参与推理部署与加速工具链建设,集成主流推理引擎(vLLM/SGLang/TensorRT-LLM等),落地模型量化(GPTQ/AWQ/INT8)、动态批处理与KV-Cache管理等特性;
5、设计并实现统一的Python/CLI SDK 与MLOps流水线,串联数据预处理、模型训练、评测、转换、部署全生命周期,建设模型版本管理与实验追踪工具;
6、与 AI Infra 工程师协作,将底层复杂的调度、存储、网络能力抽象为算法工程师“开箱即用”的 API,沉淀最佳实践文档,提升团队整体交付效率。
招聘要求:
1、硕士及以上学历,计算机类、电子信息类、软件工程、人工智能、自动化等相关专业;
2、扎实的计算机基础,掌握数据结构与算法、操作系统、计算机网络等核心知识;熟练掌握 Python,熟悉异步/并发编程,具备良好的代码规范,了解c++、cuda编程;
3、熟悉 Linux 环境与容器化基本功,掌握 Docker 基本使用、Shell 脚本及常见调试排障方法;理解异步任务队列、限流/重试/熔断等分布式系统基本概念;
4、对以下任一方向有课程项目、竞赛或实习经历:
训练优化方向:熟悉分布式训练并行策略(DP/TP/PP/ZeRO),了解大模型显存管理与通信机制(NCCL),有 Megatron/DeepSpeed 使用经验;
推理加速方向:熟悉主流推理引擎架构与使用,了解模型量化、剪枝、蒸馏或 KV-Cache 优化原理;
MLOps工具链方向:有模型训练、评测、转换、部署流水线开发经验,熟悉 MLflow / Wandb 等实验追踪工具; 5、加分项(任一即可):1. 了解 KV Cache、投机采样等 LLM 推理优化技术;2. 有 CUDA/Triton 算子编写与性能调优经验;3.参与过百亿/千亿参数大模型从 0 到 1 的训练或高并发推理部署落地;4.有自动驾驶(感知/规划/决策)算法工程化经验;5.有开源贡献/技术博客/完整工程作品集(GitHub/Gitee); 6、具备良好的沟通协作与文档能力,主动性强,愿意跨团队合作落地 AI 能力。
公司介绍
零
零跑汽车有限公司
零跑汽车有限公司,成立于2017-01-12,位于浙江,法定代表人:朱江明,登记状态:存续,注册资本470800万人民币,公司类型:民营/私企
公司信息:公司名称:零跑汽车有限公司公司类型:机械制造/机电/重工公司性质:民营企业公司地点:浙江省金华市公司规模:5000以上涉及领域:--公司网址:--