高性能异构计算高级工程师

查看原岗位

地点：上海薪资：30-60K 日期：2026-06-26

岗位摘要

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

熟悉C++泛型编程与设计模式，3年以上C++使用经验或2年高性能计算开发经验
良好的异构编程基础，熟悉NV或AMD架构，异构编程语言CUDA或HIP
熟悉Tensor-Core算子MMA/Attention的开发，拥有cutlass/flash-attention/FlashMLA/FlashInfer等项目开发经验优先
了解Pytorch/vLLM/TensorRt-llm/Deepseek等DL框架和大模型基础工程，有开发使用经验优先

上海闵行区临港浦江科技广场16号楼