太初(无锡)电子科技有限公司招聘AI芯片推理框架研究员
招聘职位:
AI芯片推理框架研究员 搜索同类职位岗位职责:1、负责自研 AI 芯片框架的架构设计与核心研发。2、实现计算图 IR 及图优化 Pass(算子融合、内存复用、并行调度等)。3、开发多后端代码生成与自动调优能力(如 AutoTVM/Ansor)。4、完成主流框架(PyTorch/TensorFlow/ONNX)模型到编译框架的转换与部署,支持 LLM 等模型的编译优化。任职条件:1、硕士及以上学历,计算机、电子、自动化等相关专业,有深度学习编译或推理引擎开发研究经验优先。2、精通 C++/Python,熟悉计算机体系结构、并行编程及内存模型。3、深入理解图优化技术(算子融合、内存规划、并行调度)及编译器 Pass 机制。4、具备性能分析与调优能力,能利用 profiling 工具定位编译后 kernel 的瓶颈。5、良好的团队协作与沟通能力,能推动技术方案落地。加分项:* 熟悉 LLM、TVM、MLIR、XLA 等至少一种开源 AI 编译框架,并有深度优化或代码贡献经验。* 了解自动调优(AutoTVM/Ansor)或张量化编译(Polyhedral 模型)。* 熟悉常见模型(ResNet、BERT、YOLO、LLaMA 等)的推理优化方法(量化、剪枝、蒸馏)。* 有 CUDA/OpenCL/汇编优化或算子库开发经验。* 有 LLM 推理优化经验(如动态 shape、PagedAttention、FlashAttention 等)。
此信息由复旦大学 (查看来源)审核并发布,我们转载该信息,仅出于传递更多就业招聘资讯、促进大学生及广大求职者就业之目的。该招聘职位信息的真实性、准确性、时效性及合法性均由原始发布方“复旦大学”负责。我们作为信息转载平台,不构成求职建议,不涉及任何职业中介服务,不对其内容承担任何形式的保证责任。请用户在使用转载信息时保持审慎,自行判断并承担相应风险,求职请认准企业官方渠道!