Job Description

CUDA优化专家

北京

社招

全职

互联网 / 电子 / 网游 - 研发

职位 ID：A190471

职位描述

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. 统筹协调能力强，有大型研发团队管理经验优先。

职位信息

部门： 推理计算

投递