小鹏集团
【27届校招】AI Infra 工程师(模型训练与推理加速方向)
岗位职责
小鹏汽车是中国领先的智能电动汽车公司之一。我们设计、研发、制造并销售深度融合先进互联网、人工智能与自动驾驶技术的智能电动汽车。我们坚持自主研发与智能制造,致力于为客户创造更好的出行体验。我们努力以技术和数据变革智能电动汽车,塑造未来的出行体验。 我们正在招聘一名全职机器学习工程师(AI 基础设施方向),要求具备深厚的专业知识,并对构建业界领先的机器学习基础设施、训练超大规模基础模型、加速模型训练与推理充满热情。 我们的使命是解决自动驾驶问题。你将与一支由优秀软件工程师、机器学习工程师和研究科学家组成的团队共同工作,不断突破前沿机器学习模型的边界,赋能下一代端到端自动驾驶解决方案。 岗位职责:
设计并实现训练数据管线,从百万级车队产生的数百 PB 有标注及无标注数据中进行流式读取。
为小鹏所有物理 AI 基础模型(包括 VLA 2.0、XWorld、机器人等)实现训练框架。
运用前沿并行技术(如 FSDP、专家并行、上下文并行)及数据类型优化加速训练。
加速云端模型推理,服务于闭环仿真、强化学习以及企业级 LLM/VLM 应用。
任职要求
基本技能要求
计算机科学/计算机工程/电子工程或相关专业硕士学位,或具备同等的行业经验。
精通 PyTorch。
熟悉模型推理框架(如 vLLM、SGLang)。
深入理解 Transformer 架构及其训练和推理加速方法。
具备大规模分布式模型训练经验。
具有对模型进行性能分析并深入排查、提升训练和推理速度的实际业绩。 优先技能要求:
有自动驾驶行业相关经验。
有使用 CUDA 编写自定义算子的经验。
有边缘计算系统相关经验。
熟悉分布式计算框架,如 Ray。
有在较大团队中高效协作解决复杂问题的实际业绩。 我们提供:
充满乐趣、互相支持、积极投入的工作环境
支撑你工作的基础设施与算力资源
与业内顶尖人才共同从事前沿技术工作的机会
通过推进自动驾驶,为交通出行变革做出重大贡献的机会
具有竞争力的薪酬待遇
信息来自企业官方招聘渠道
OfferSeek 对公开岗位信息进行聚合、去重和结构化整理,最终申请以企业官方页面为准。