Job Description

【27届校招】AI Infra 工程师（模型训练与推理加速方向）

上海、广州、北京

正式

互联网 / 电子 / 网游

职位描述

小鹏汽车是中国领先的智能电动汽车公司之一。我们设计、研发、制造并销售深度融合先进互联网、人工智能与自动驾驶技术的智能电动汽车。我们坚持自主研发与智能制造，致力于为客户创造更好的出行体验。我们努力以技术和数据变革智能电动汽车，塑造未来的出行体验。 我们正在招聘一名全职机器学习工程师（AI 基础设施方向），要求具备深厚的专业知识，并对构建业界领先的机器学习基础设施、训练超大规模基础模型、加速模型训练与推理充满热情。 我们的使命是解决自动驾驶问题。你将与一支由优秀软件工程师、机器学习工程师和研究科学家组成的团队共同工作，不断突破前沿机器学习模型的边界，赋能下一代端到端自动驾驶解决方案。 岗位职责： 1、设计并实现训练数据管线，从百万级车队产生的数百 PB 有标注及无标注数据中进行流式读取。 2、为小鹏所有物理 AI 基础模型（包括 VLA 2.0、XWorld、机器人等）实现训练框架。 3、运用前沿并行技术（如 FSDP、专家并行、上下文并行）及数据类型优化加速训练。 4、加速云端模型推理，服务于闭环仿真、强化学习以及企业级 LLM/VLM 应用。

职位要求

基本技能要求： 1、计算机科学/计算机工程/电子工程或相关专业硕士学位，或具备同等的行业经验。 2、精通 PyTorch。 3、熟悉模型推理框架（如 vLLM、SGLang）。 4、深入理解 Transformer 架构及其训练和推理加速方法。 5、具备大规模分布式模型训练经验。 6、具有对模型进行性能分析并深入排查、提升训练和推理速度的实际业绩。 优先技能要求： 1、有自动驾驶行业相关经验。 2、有使用 CUDA 编写自定义算子的经验。 3、有边缘计算系统相关经验。 4、熟悉分布式计算框架，如 Ray。 5、有在较大团队中高效协作解决复杂问题的实际业绩。 我们提供： 1、充满乐趣、互相支持、积极投入的工作环境 2、支撑你工作的基础设施与算力资源 3、与业内顶尖人才共同从事前沿技术工作的机会 4、通过推进自动驾驶，为交通出行变革做出重大贡献的机会 5、具有竞争力的薪酬待遇

投递