岗位描述
岗位职责
1、AI芯片的深度学习算子开发和优化;
2、AI芯片编译工具链测试开发;
岗位要求
1.学历与专业背景:计算机、软件工程、自动化或相关专业重点本科及以上学历,具备扎实的计算机科学基础。
2. 编程与底层能力:精通 C/C++ 编程,熟练掌握模板元编程;深入理解 GPU 体系结构(SM、Warp、内存层级等),了解 CUDA 编程模型。
3.对高性能计算库:掌握 CUTLASS、CUTE、CUB 等计算库的底层原理与源码,有基于这些库开发过高性能算子(如 GEMM、FlashAttention)的实战经验优先。
4. 有PyTorch和Triton开发经验优先;