AI研究算法工程师-高性能算子开发方向(J13373)

AI 摘要智能生成

通信算子研发与调优 - 针对跨卡

职位描述

本岗位聚焦AI算力核心算子研发，致力于为AI训练、推理场景打造高性能算子与通信方案，深度联动硬件特性与软件优化，为业务落地提供坚实的算力支撑，高效释放硬件理论潜能。 1、算子开发与优化 - 围绕英伟达GPU、华为NPU、海光DCU等主流硬件架构，依托编程语言和计算架构，完成大模型核心计算和通信算子研发和优化。 - 立足芯片存储层级、计算单元、指令集特性，开展针对性性能调优，确保算子性能精准匹配硬件能力，为业务提供高效算力支持。 2、通信算子研发与调优 - 针对跨卡、跨机等多样化通信场景，设计、开发并优化通信算子，保障分布式训练、推理的高效通信链路。 3、协同与优化 - 紧跟模型结构与算法迭代趋势，协同框架、算法团队，设计适配硬件特性的模型方案，挖掘硬件与算法的协同潜力。 - 围绕真实训练、推理负载开展性能分析，为模型落地提供性能优化支撑，保障业务高效运行。

任职要求

• 算法