岗位描述
岗位职责
岗位职责:
1.研发Triton/TileLang等DSL编译框架;
2.扩展DSL语法并开发图优化Pass
3.实现算子融合与全局内存规划
4.研发自研芯片后端CodeGen与调度;
5.汇编级指令调优对标手写性能
6.对接Triton生态与开源社区
岗位要求
岗位要求:
1.计算机相关专业本科及以上;
2.精通C++/Python,熟悉Linux;
3.熟悉GPGPU/NPU架构及并行模型;
4.深入理解MLIR/LLVM及Pass开发;
5.责任心强,具备问题闭环意识;
6.具备图优化与底层CodeGen经验加分;Triton kernel 开发或编译器修改;LLVM / MLIR pass 开发;CUDA kernel 性能优化,理解 PyTorch 编译机制等加分项;
您的工作地点可以base 上海,北京,成都,西安等地