岗位描述
岗位职责
小鹏汽车是中国领先的智能电动汽车公司之一。我们设计、研发、制造并销售深度融合先进互联网、人工智能与自动驾驶技术的智能电动汽车。我们坚持自主研发与智能制造,致力于为客户创造更好的出行体验。我们努力以技术和数据变革智能电动汽车,塑造未来的出行体验。
我们正在招聘一名全职机器学习工程师(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、具有竞争力的薪酬待遇