词元星火-算子开发-北京(J13593)

职位描述

1、 算子开发与优化

- 围绕GPU、NPU等主流硬件架构，依托CUDA、Ascend C、TileLang等技术，完成大模型核心计算和通信算子研发和优化。

- 立足芯片存储层级、计算单元、指令集特性，开展针对性性能调优，确保算子性能精准匹配硬件能力，为业务提供高效算力支持。

通信算子研发与调优

- 针对跨卡、跨机等多样化通信场景，设计、开发并优化通信算子，保障分布式训练、推理的高效通信链路。

- 持续对NCCL、HCCL等主流集合通信库进行性能打磨与故障排查，保障通信环节稳定可靠，适配大规模算力协同需求。

2、协同与优化

- 紧跟模型结构与算法迭代趋势，协同框架、算法团队，设计适配硬件特性的模型方案，挖掘硬件与算法的协同潜力。

- 围绕真实训练、推理负载开展性能分析，为模型落地提供性能优化支撑，保障业务高效运行。

任职要求

1、基础门槛

- 研究生及以上学历。

- 编程功底扎实，精通C++/C语言，具备良好的代码规范意识与数据结构、算法基础，熟悉操作系统核心原理。

- 熟练使用Git等代码管理工具，养成规范的开发习惯；熟练掌握linux下开发能力。

- 热爱性能优化，兼具技术视野与代码品质追求，具备良好的中文沟通能力、团队协作意识、责任心与自驱力，学习能力突出。

2、技术必备

- 熟练掌握CUDA、Ascend C等编程原理，能够基于编程语言开展算子开发工作。

- 掌握编译优化技术，清晰理解GPU/NPU架构的并行执行、内存层级与计算单元特性。

- 熟练掌握TileLang、Triton等编译原理及使用，编写和优化核心算子。

- 对硬件底层架构有清晰认知，能够结合硬件特性开展针对性优化。

加分项

1、算子专长：有CUDA或Ascend C算子开发经验，熟悉LLM相关算子等算子库，深入理解AI加速卡硬架构和特性。

2、竞赛背景：在高性能计算相关赛事取得较好成绩。

3、行业经验：拥有3年以上算子库优化经验，执着于挖掘硬件性能、提升软件效率。