AI推理工程师-无锡(J10425)

职位描述

1.负责AI推理/训练框架（如PyTorch、TensorFlow等）在自研芯片上的适配、扩展与深度优化。

2.设计与实现模型调度、算子融合及内存管理机制，解决大模型在异构硬件上的部署瓶颈。

3.打通从AI框架到底层驱动的完整软件栈，构建高性能、易用的端到端开发环境。

4.跟踪业界前沿框架技术，结合自研芯片微架构特性，推动软硬件协同设计方案的落地

任职要求

1.编程能力：精通C/C++及Python编程，具备良好的代码风格和面向对象设计能力，熟悉常用数据结构与算法。

2.框架经验：深入理解PyTorch或TensorFlow等主流深度学习框架的底层原理（如自动求导、动态图/静态图机制、分布式训练架构），有框架源码阅读或二次开发经验者优先。

3.系统知识：熟悉编译器基本原理（前端、IR、后端），了解ONNX、TVM等相关技术栈；熟悉Linux操作系统及多线程/多进程编程。

加分项：

1.有AI编译器开发经验；

2.有LLM（大语言模型）推理框架（如vLLM、TensorRT-LLM）优化经验；

3.有CUDA/OpenCL编程经验，熟悉GPU/NPU等异构计算架构。