公司为影微创新,是商汤科技的子公司,专注于做端侧AI芯片,目前已流片。网上有不少关于影微的
公开信息。
### 职位名称:AI工具链工程师(自研芯片模型编译方向)
工作地点: 北京、上海、西安、成都、深圳
职位类型: 全职
#### 岗位职责:
1. 负责开发面向自研芯片的AI模型编译工具链,支持CNN、Transformer等主流网络的高效部署。
2. 设计并实现模型优化、图编译、算子融合、内存优化等关键技术,提升芯片上的推理性能。
3. 开发自动化工具链,简化模型从训练框架(如PyTorch/TensorFlow)到自研芯片的部署流程。
4. 与芯片团队紧密合作,针对硬件特性(如指令集、内存架构)进行深度优化。
5. 负责工具链的API设计、文档建设和开发者体验优化,推动工具链在客户侧的落地。
#### 任职要求:
必需条件:
- 计算机、电子工程、数学等相关专业硕士及以上学历,1年以上AI工具链开发经验。
- 精通Python和C++,具备高性能代码开发和调试能力。
- 熟悉AI模型部署全流程,包括模型量化、剪枝、编译优化等。
- 深入理解CNN/Transformer等网络结构及其计算特性。
- 有编译器开发经验(如LLVM、MLIR、TVM等),熟悉中间表示(IR)优化。
- 熟悉至少一种主流深度学习框架(PyTorch/TensorFlow/MXNet)。
加分项:
- 有TVM开发经验,熟悉TVM Stack(Relax/TIR/Ansor)。
- 有AI算子开发经验,熟悉CUDA/OpenCL或芯片特定指令集优化。
- 参与过完整AI工具链开发项目,产出过成熟部署工具。
- 了解自研芯片架构(如NPU、DSA),能针对硬件特性优化工具链。
#### 我们提供:
- 参与前沿AI芯片研发的机会,直接影响工具链的技术方向。
- 技术挑战与成长空间:解决高并发、低延迟、异构计算等难题。
- 有竞争力的薪资+期权/股票+完善的福利体系。
薪资待遇:30k~60k * 15月,具体视面试表现和经历。
有意者欢迎发送简历到 aisecurity@sensetime.com
--
修改:segfault FROM 124.126.230.*
FROM 124.126.230.*