资深模型推理优化工程师

职位描述

岗位职责

1. 负责智能驾驶模型在多种不同硬件平台的极致性能优化，结合编译优化、并行计算优化、图融合、高效 CUDA 算子开发实现行业领先的车端推理性能。

2. 针对特定NPU硬件计算平台，基于对硬件体系结构的深入理解，实现对硬件的高效利用。

3. 针对Pytorch/CUDA相关GPU计算任务，进行算子和系统优化，提升训练/推理效率。

任职要求

岗位要求

1. 工作3年以上，计算机/数学/物理/电子工程/自动控制专业硕士及以上学历。

2. 优秀的代码能力，熟练掌握C/C++或Python，有 CUDA 开发经验，熟悉TVM/TensorRT/Triton/Cutlass等推理框架优先。

3. 熟悉计算机体系结构，对GPU、NPU等有比较深刻的理解。

4. 熟悉CNN/Transformers/DETR等主流模型原理。

5. 良好的团队协作能力；创新性强，有良好的动手实现能力，对技术有热情。