返回岗位列表

GPU芯片高性能软件栈开发工程师

查看原岗位

地点：上海薪资：薪资未公开日期：2026-01-29

岗位摘要

参与壁仞GPU芯片高性能软件栈的开发、优化和文档编写；负责高性能算子算法的设计、分析、评估与优化；参与硬件关键功能、精度和性能的研讨与评估；关注社区关于芯片和大模型的最新动态，探索前沿技术方向在壁仞芯片上的落地

岗位职责

参与壁仞GPU芯片高性能软件栈的开发、优化和文档编写
负责高性能算子算法的设计、分析、评估与优化
参与硬件关键功能、精度和性能的研讨与评估
关注社区关于芯片和大模型的最新动态，探索前沿技术方向在壁仞芯片上的落地
负责软件栈（如Sutlass/SCCL）的迭代开发、优化和文档编写，以完善壁仞生态软件栈
负责TCore/CUB相关算子算法的设计、分析和评估，并与编译器团队合作给出优化方案，推动性能达标
在验证截断阶段发掘硬件在关键功能、精度和性能上的问题
探索前沿技术方向在壁仞芯片上的落地应用

任职要求

熟悉C++泛型编程与设计模式，拥有3年以上C++使用经验或2年高性能计算开发经验
具备良好的异构编程基础，熟悉NV或AMD架构，以及异构编程语言CUDA或HIP
了解Pytorch、vLLM、TensorRt-llm、Deepseek等DL框架和大模型基础工程，有开发使用经验者优先