软件工程师 - pytorch训练框架国产芯片适配

小马智行 · 北京

来源：小马智行官方招聘 · 更新于 2026-06-04 · 抓取快照，投递前请确认岗位仍在招

全职技术北京

岗位描述

将前沿的端到端自动驾驶等模型训练，在国产GPU训练芯片上进行适配，并管理大规模训练集群。在这个岗位上，你会负责： 1. 把基于Nvidia GPU的pytorch训练，迁移到阿里平头哥等国产训练GPU上； 2. 把基于CUDA实现的自定义算子，迁移到国产训练GPU上； 3. 把pytorch中的一些复杂算子，迁移到国产训练GPU上； 4. 深入分析国产GPU训练芯片的设计和软件接口，充分挖掘GPU性能； 5.… 1. 有把pytorch训练任务，从Nvidia迁移到阿里平头哥等某一家国产训练GPU的经验； 2. 精通某一家国产训练GPU训练框架； 3. 熟悉GPU设计理念，熟悉CUDA编程…