Ivyea Jobs 6451 roles · 88 companies · 刚刚
昆仑芯 · AI 芯片

AI推理框架研发工程师

北京 社招 · 全职 软件研发类 推理 / 部署优化

职位描述

1、负责AI模型对昆仑芯AI芯片的适配和性能优化工作,完成客户需求的落地任务 2、为昆仑芯AI芯片各系列高性能加速芯片提供推理软件栈,包括框架,图编译器以及周边产品的技术落地 3、支持主流框架的模型对接,算法优化,组网开发,算子开发,上线部署,并探索新一代AI芯片编程模型和架构 4、昆仑芯AI芯片技术栈相关的研发工作

任职要求

1、计算机及计算机相关专业,硕士及以上学历 2、熟悉Python/C/C++,良好的编程习惯,熟悉计算机基础算法和数据结构,熟悉 Linux 开发环境 3、熟悉深度学习相关知识,有框架(vllm、sglang、Pytorch,PaddlePaddle,TVM,MLIR,ONNX等)开发或使用经验优先 4、有模型推理、性能调优,大模型等等经验优先