词元星火-算子开发-合肥(J13592)

AI 摘要智能生成

围绕GPU、NPU等主流硬件架构，依托CUDA

职位描述

1、 算子开发与优化 - 围绕GPU、NPU等主流硬件架构，依托CUDA、Ascend C、TileLang等技术，完成大模型核心计算和通信算子研发和优化。 - 立足芯片存储层级、计算单元、指令集特性，开展针对性性能调优，确保算子性能精准匹配硬件能力，为业务提供高效算力支持。 通信算子研发与调优 - 针对跨卡、跨机等多样化通信场景，设计、开发并优化通信算子，保障分布式训练、推理的高效通信链路。 - 持续对NCCL、HCCL等主流集合通信库进行性能打磨与故障排查，保障通信环节稳定可靠，适配大规模算力协同需求。 2、协同与优化 - 紧跟模型结构与算法迭代趋势，协同框架、算法团队，设计适配硬件特性的模型方案，挖掘硬件与算法的协同潜力。 - 围绕真实训练、推理负载开展性能分析，为模型落地提供性能优化支撑，保障业务高效运行。

任职要求

• 算法