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等异构计算架构。

