飞腾信息技术有限公司
AI算子优化工程师(GPU/CUDA方向)
RESPONSIBILITIES
岗位职责
负责 AI 大模型(如 Transformer 类、多模态模型)在自研芯片上的性能极致优化,覆盖训练与推理全流程;
深入 AI 框架层(PyTorch/TensorFlow)与 CUDA 软件栈,进行算子开发、内核融合及内存/通信优化;
主导关键模型(如 LLM、Diffusion)的分布式训练与推理性能调优,突破算力与通信瓶颈;
构建芯片级算子性能分析工具链,推动软硬件协同设计迭代。
REQUIREMENTS
任职要求
计算机/电子/数学等相关专业本科及以上学历,3 年以上 AI 底层开发经验;
精通 CUDA 编程与 GPU 性能优化,有算子库(如 cuDNN、CUTLASS)开发或调优经验;
熟练掌握 PyTorch/TensorFlow 框架机制,具备模型训练/推理部署实战经验;
具备扎实的 C++/Python 能力,熟悉 Linux 开发环境及容器化工具链。 薪资: 面议 学历要求(及以上): 本科 工作经验: 5-8年 年龄要求: 不限 其他要求:
加分项
有 LLM、视频生成等前沿模型优化经验,或参与过开源 AI 框架贡献;
熟悉芯片架构(如 SIMT、内存层次结构),有 AI 芯片开发经验者优先;
发表过高性能计算/体系结构相关顶会论文或专利。
信息来自企业官方招聘渠道
OfferSeek 对公开岗位信息进行聚合、去重和结构化整理,最终申请以企业官方页面为准。