岗位描述
工作职责:
1.负责tilelang等DSL编译器开发工作,适配自研AI芯片;
2.芯片指令集在自研芯片上的功能验证及性能调优;
3.适配/优化基于tilelang的高性能算子库;
4.与硬件团队合作,共同优化芯片架构及性能。
任职要求:
1.计算机相关专业,精通C++及python;
2.熟悉主流的DSL编程语言(triton/tilelang),有triton/tilelang编译器开发经验;
3.熟悉MLIR(或者TVM)、以及LLVM编译器框架;
4.熟悉GPGPU架构及cuda,精通体系结构及编译原理。
5. 了解agent kernel编程的方法。
具有以下条件者优先:
1、计算机领域相关的编程大赛获奖者优先;
2、熟悉算子开发和优化方法者优先;
3、熟悉MLIR, TVM, XLA, LLVM优先。
4、在国产芯片厂商(寒武纪,璧仞,燧原,华为,沐曦,天数)有性能优化经验优先