职位&公司对比

职位详情

• 上海
• 不限
• 本科

• Python
• C++
• TensorFlow
• PyTorch

岗位职责： 1、从事triton算子的性能优化工作； 2、分析模型性能瓶颈，针对特定异构硬件平台（GPU/DSA）进行高性能计算加速和优化。 3、承担deeplink triton编译器开发，包括相关优化pass开发以及多后端(GPU/DSA)对接 岗位要求： 1、熟悉cuda/华为昇腾AscendC/triton编程语言； 2、熟悉C++/Python，编程语言基础扎实； 3、熟悉计算机体系结构及操作系统来自BOSS直聘原理； 4、了解kanzhun深度学习算法，对大模型推理和训练框架vllm megatron的应用或开发经验; 5、优秀的分析问题和解决问题的能力。 具有以下条件者优先： 1、计算机领域相关的编程大赛获奖者优先； 2、熟悉AI领域常见框架、模型，有过实战调优经验者优先； 3、熟悉大模直聘型算法和优化方法者优先； 4、熟悉MLIR, TVM, XLA, LLVM优先。 5、在国产芯片厂商(寒武纪，璧仞，燧原，华为，沐曦，天数)性能优化经验优先