职位描述
1. 负责带领团队建设CUDA研发体系和GPU底层优化技术体系,跟踪和探索面向大模型的GPU底层优化技术新趋势。
2. 负责带领CUDA研发团队完成大模型算子计算库的 CUDA 开发及Kernel优化。
3. 负责与大模型工具链团队一起,共同完成大模型在云-边-端应用场景的压缩加速和部署优化。任职要求
1. 研究生及以上学历,计算机/电子相关专业优先,至少3—5年CUDA开发经验。
2. 熟悉常见CUDA优化手段,包括Memory Coalescing、Bank Conflict、Double Buffer、Tile Size Tuning、Tensor Core、Register Reuse等。
3. 熟悉常用CUDA调试技术,熟悉Profiling中常见Metric分析以及流水线Stall分析。
4. 使用CUDA源代码写出过性能接近或超过cuBLAS/CUTLASS计算库代码的候选人优先。
5. 使用CUTLASS开发过高性能融合算子的候选人优先。
6. 统筹协调能力强,有大型研发团队管理经验优先。