返回岗位列表

壁仞科技

异构计算软件工程师

地点:上海 薪资:薪资未公开 日期:2025-09-25

岗位摘要

负责软件栈(如Sutlass/SCCL)的迭代开发、优化和文档编写,以完善壁仞生态软件栈;负责TCore/CUB相关算子算法的设计、分析和评估;与编译器团队合作,为算子算法提供优化方案,推动性能达标

岗位职责

  • 负责软件栈(如Sutlass/SCCL)的迭代开发、优化和文档编写,以完善壁仞生态软件栈
  • 负责TCore/CUB相关算子算法的设计、分析和评估
  • 与编译器团队合作,为算子算法提供优化方案,推动性能达标
  • 在验证过程中,发掘硬件在关键功能、精度和性能方面的问题
  • 探索前沿技术方向在壁仞芯片上的落地应用

任职要求

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