AI编译器模型部署工程师
岗位信息
| 招聘单位 | 小鹏汽车 |
|---|---|
| 工作地点 | 上海 |
| 官方更新时间 | 2024-09-05 20:31:05 |
职位描述
工作职责:
主要负责AI编译器开发,模型部署和性能优化。
1. 实现基于NPU/GPU架构的AI编译器开发;
2. 模型适配和部署;
3. 基于模型及硬件特性进行性能优化。
任职要求
任职要求:
1.本科及以上, 有良好的计算机理论基础,熟悉体系结构者优先,包括GPU,CPU,NPU,DSP等
2.优秀的C/C++编程能力, 熟悉cuda以及pytorch等优先
3.有框架和算子开发经验者优先, 包括TensorRT,pytorch,tensorflow,mindspore,tvm,mlir,oneflow,paddle等等;
4.编译优化经验:具备 MLIR/LLVM/TVM/XLA/clang等编译优化经验,熟悉AI编译软件栈的设计与实现
6.问题解决能力:具备出色的问题分析与解决能力,能够针对复杂的系统架构提出有效的优化方案
7.沟通能力:与人沟通流畅,能准确表达自己观点
提示:投递请认准招聘单位官方招聘官网,谨防中介收费。