以上为北京时间。链接检查不代表仍在招聘,申请前请以招聘官网为准。
按这个岗位的城市和方向提醒,确认后才开启,可选择每日或每周。
岗位职责:1、参与AI编译器前端/调度/后端的设计与开发,实现芯片高性能算子库;2、参与AI模型的图优化、图拆分、算子融合、调度优化、指令生成等;3、协同芯片、框架、驱动等团队,分析和优化编译器性能;4、编写高质量、可测试、可维护的代码。
岗位职责:
1、参与AI编译器前端 / 调度 / 后端的设计与开发,实现芯片高性能算子库;
2、参与AI模型的图优化、图拆分、算子融合、调度优化、指令生成等;
3、协同芯片、框架、驱动等团队,分析和优化编译器性能;
4、编写高质量、可测试、可维护的代码,并输出高质量文档。
任职要求:
1、计算机、电子工程、自动化、人工智能等相关专业,研究生及以上学历;
2、熟悉编译原理,掌握数据流图、中间表示等基础知识;
3、熟悉LLM与CNN模型,了解常见算子的详细计算方法;
4、熟悉 C++/Python 编程,具备良好的编码能力;
5、有 LLVM、MLIR、TVM、TensorRT、XLA 等相关编译器项目经验者优先;
6、有 AI 框架(如 PyTorch、TensorFlow、ONNX)使用或对接经验者加分;
7、具备良好的学习能力、沟通能力和团队协作意识。
加分项:
1、有大模型部署、推理加速经验;
2、了解自动调优、算子模板生成、张量编程等概念。
软件和信息技术服务业