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

1、参与壁仞GPU芯片高性能软件栈的开发，优化和文档编写

2、负责高性能算子算法的设计、分析、评估与优化

3、参与硬件关键功能、精度和性能研讨与评估

4、关注社区关于芯片和大模型的最新动态，探索前沿技术方向在壁仞芯片上的落地

1、熟悉C++泛型编程与设计模式，3年以上C++使用经验或2年高性能计算开发经验

2、良好的异构编程基础，熟悉NV或者AMD架构，异构编程语言CUDA或HIP

3、熟悉Tensor-Core算子MMA/Attention的开发， 拥有cutlass/flash-attention/FlashMLA/FlashInfer等项目开发经验优先

4、了解Pytorch/vLLM/TensorRt-llm/Deepseek等DL框架和大模型基础工程，有开发使用经验优先

核心产出：

1、软件栈Sutlass/SCCL的迭代开发，优化和文档编写，完善壁仞生态软件栈

2、TCore/CUB相关算子算法的设计、分析和评估，并跟编译器团队合作给出优化方案，推动性能达标

3、在验证截断发掘硬件关键功能、精度和性能上的问题

4、探索前沿技术方向在壁仞芯片上的落地