AI编译器开发工程师（上海苏州杭州成都）

该职位来源于猎聘 职位描述 1. 参与存算一体架构芯片的AI编译器框架设计和开发，支持TVM/MLIR/Triton等开源工具链的适配与优化，实现从AI模型到存算芯片的高效编译; 2. 基于自研存算架构，开发图级别调度、算子融合、内存管理等算法，优化计算图在存算芯片上的执行效率; 3. 参与PyTorch等主流框架的适配，以及torch.compile等编译框架的适配; 4. 开发模型压缩和分布式编译工具，优化大模型在存算芯片上的高效推理; 职位要求： 1.计算机/电子/数学相关专业硕士及以上学历，2年或以上AI编译器或高性能计算开发经验，熟悉TVM/MLIR/torch.compile等至少一种编译框架，有实际优化案例; 2.精通C++/Python，熟悉Linux开发环境，掌握Shell脚本和调试工具（如GDB），具备出色的程序开发、设计和调试能力; 3.良好的沟通协作和解决问题能力、时间管理能力、自我驱动能力。