GPGPU架构师
岗位摘要
参与壁仞新一代GPGPU架构的研发工作，负责计算单元、存储体系以及高性能互连架构的设计与优化；基于壁仞GPGPU架构，针对AI大模型进行端到端的性能优化，涵盖模型训练与推理的全流程，提升算子效率与系统吞吐
岗位职责
- 参与壁仞新一代GPGPU架构的研发工作，负责计算单元、存储体系以及高性能互连架构的设计与优化
- 基于壁仞GPGPU架构，针对AI大模型进行端到端的性能优化，涵盖模型训练与推理的全流程，提升算子效率与系统吞吐
- 与软件、算法及硬件实现团队紧密合作，解决跨层级的性能瓶颈问题
- 跟踪GPGPU与AI计算领域的最新技术趋势，参与前沿研究，提出创新性架构解决方案，并推动技术成果的落地与应用
任职要求
- 计算机科学、微电子、电子工程或相关专业博士或硕士学位，具备3年以上GPGPU架构设计与优化相关经验
- 深入理解GPU/NPU体系结构，包括计算单元设计、缓存层次结构（L1/L2 Cache）、内存带宽优化及高性能互连架构的设计原理与实现细节
- 熟悉现代GPGPU架构（如NVIDIA Ampere、Hopper架构）的设计理念
- 深入理解AI及GPGPU算子实现原理，熟悉大模型（如Transformer、Diffusion模型）的架构设计及PyTorch、TensorFlow等框架
- 精通GPU指令集架构，能够基于指令级并行性进行深度优化，具备基于PTX/SASS进行指令级算子优化的实战经验
- 精通CUDA等并行编程模型，能够针对特定硬件特性进行精细调优
- 熟练掌握C/C++和Python等常用编程语言
- 深入掌握PyTorch、TensorFlow等主流AI框架，熟悉CUTLASS等高性能计算库，能够进行算子开发与优化
- 具备熟练的英文读写能力，能够无障碍阅读技术文献、撰写技术报告
- 在顶级学术会议（如ISCA、MICRO、HPCA）或期刊上发表过相关领域论文者优先
- 具备成功流片经验者优先
福利待遇
- 薪资范围：50,000–80,000元/月，共15薪
- 工作地点：上海闵行区壁仞科技
工作地址
上海闵行区壁仞科技