职位名称
AI编译器工程师(校园招聘,工作地点无锡/西安)
单位名称
北极雄芯信息科技(西安)有限公司
任职要求
计算机、电子工程、自动化、人工智能等相关专业,研究生及以上学历; 熟悉编译原理,掌握数据流图、中间表示等基础知识; 熟悉LLM与CNN模型,了解常见算子的详细计算方法; 熟悉C++/Python编程,具备良好的编码能力; 有LLVM、MLIR、TVM、TensorRT、XLA等相关编译器项目经验者优先; 有AI框架(如PyTorch、TensorFlow、ONNX)使用或对接经验者加分; 具备良好的学习能力、沟通能力和团队协作意识。 有大模型部署、推理加速经验; 了解自动调优、算子模板生成、张量编程等概念。
岗位职责
参与AI编译器前端/调度/后端的设计与开发,实现芯片高性能算子库; 参与AI模型的图优化、图拆分、算子融合、调度优化、指令生成等; 协同芯片、框架、驱动等团队,分析和优化编译器性能; 编写高质量、可测试、可维护的代码,并输出高质量文档。