Ivyea Jobs 6505 roles · 88 companies · 刚刚
无问芯穹 · AI Infra

CUDA优化专家

北京 社招 · 全职 推理 / 部署优化

职位描述

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. 统筹协调能力强,有大型研发团队管理经验优先。
无问芯穹去官方页面投递 →