职位编号 J20251222170800084687
AI编译器工程师平台直营HC多
60-100万北京市本科5-8年招聘 10 人预计佣金 97.5K23小时前刷新/6月4日 11:12重新发布
职位信息项目手册职位进展做单须知佣金规则发单顾问
寻访须知
必要条件: • 熟练掌握C/C++/Python等编程技能,有扎实的编程基础; • 熟悉GPU上算子编程与优化(如CUDA,Cutlass, Cute, Triton, TileLang等) • 熟悉一项或多项AI编译器技术或AI框架:如Torch.compiler, Triton, TileLang, MLIR, TVM, XLA, Pytorch, TensorFlow, ONNX等 • 对主流模型架构及相关算子优化技术熟悉者优先考虑 • 对NCCL或类似通信编程技术熟悉者优先考虑 • 有超算或并行编程经验优先考虑(如熟悉分布式矩阵乘法、MPI等) • 有过分布式和异步编程经验优先考虑 • 有其它AI芯片的算子开发、模型优化、低精度量化、部署经验者优先级考虑
优先条件:985/211以上硕士研究生
目标公司/行业:同类半导体公司,摩尔线程,寒武纪,壁仞等
必问问题:
待发单方补充
JD基本信息
岗位职责
面向大带宽GPU架构芯片,研发新的AI编译器框架及相关技术,包括但不限于分布式编译、数据流编译、图编译、算子编译、量化优化、通算融合等,从而支持主流大模型的高效运行。
任职要求
• 熟练掌握C/C++/Python等编程技能,有扎实的编程基础; • 熟悉GPU上算子编程与优化(如CUDA,Cutlass, Cute, Triton, TileLang等) • 熟悉一项或多项AI编译器技术或AI框架:如Torch.compiler, Triton, TileLang, MLIR, TVM, XLA, Pytorch, TensorFlow, ONNX等 • 对主流模型架构及相关算子优化技术熟悉者优先考虑 • 对NCCL或类似通信编程技术熟悉者优先考虑 • 有超算或并行编程经验优先考虑(如熟悉分布式矩阵乘法、MPI等) • 有过分布式和异步编程经验优先考虑 • 有其它AI芯片的算子开发、模型优化、低精度量化、部署经验者优先级考虑
所属行业:半导体/芯片/集成电路
职能分类:芯片测试工程师
工作城市:
北京,招聘10人,详细地址:上海/北京/西安
职位要求
学历要求:本科·统招
工作年限:5-8年
薪资福利
年薪范围:60-100万 16薪
薪资福利:-
面试信息
面试轮次:2轮
视频面试:可以接受
面试流程:-