ML编译器工程师

我们正在寻找一名全职的机器学习编译器工程师,对建立先进的机器学习编译器基础设施有深入的了解和强烈的热情，以便在通用目的GPU或定制SoC上部署先进的自动驾驶模型。
【工作职责】
开发ML编译器，使机器学习模型能够在通用GPU或小鹏汽车定制的SoC芯片上运行，具体的工作包括：
1、clang/MLIR/LLVM编译器工具链开发，包括C/C++、汇编器、链接器。
2、各种前端优化，包括各种计算图优化。
3、各种后端优化，包括通过异构内存管理、分块和并行化。
4、测试和分析编译器部署的模型。

【岗位要求】
1、计算机科学/计算机工程/电气工程硕士及以上学位，或具有同等行业经验。
2、优秀的C/C++开发能力。

【加分项】
1、在自动驾驶行业有相关经验。
2、使用多级IR编译器框架的开发经验。
3、熟悉Cuda和Pytorch。
4、熟悉GPU、CPU、NPU、DSP 架构。
5、在大型团队中高效协作解决过复杂问题。