软件工程师 - pytorch训练框架国产芯片适配
职位描述
将前沿的端到端自动驾驶等模型训练，在国产GPU训练芯片上进行适配，并管理大规模训练集群。
在这个岗位上，你会负责：
1. 把基于Nvidia GPU的pytorch训练，迁移到阿里平头哥等国产训练GPU上；
2. 把基于CUDA实现的自定义算子，迁移到国产训练GPU上；
3. 把pytorch中的一些复杂算子，迁移到国产训练GPU上；
4. 深入分析国产GPU训练芯片的设计和软件接口，充分挖掘GPU性能；
5. 深入了解自动驾驶领域的各类前沿模型设计，深入分析模型训练瓶颈；
6. 管理大规模训练集群，提升GPU利用率。任职要求
1. 有把pytorch训练任务，从Nvidia迁移到阿里平头哥等某一家国产训练GPU的经验；
2. 精通某一家国产训练GPU训练框架；
3. 熟悉GPU设计理念，熟悉CUDA编程；
4. 熟悉pytorch训练框架，能够分析训练性能瓶颈；
5. 优秀的python、GPU编程能力。
加分项：
1. 有过GPU自定义算子的开发经验。
