AI算子优化工程师（GPU/CUDA方向）

招聘人数：5
工作地点：长沙、成都、上海
发布日期：2026年9月20日

岗位职责：
1. 负责 AI 大模型（如 Transformer 类、多模态模型）在自研芯片上的性能极致优化，覆盖训练与推理全流程；
2. 深入 AI 框架层（PyTorch/TensorFlow）与 CUDA 软件栈，进行算子开发、内核融合及内存/通信优化；
3. 主导关键模型（如 LLM、Diffusion）的分布式训练与推理性能调优，突破算力与通信瓶颈；
4. 构建芯片级算子性能分析工具链，推动软硬件协同设计迭代。

任职条件：
1. 计算机/电子/数学等相关专业本科及以上学历，3 年以上 AI 底层开发经验；
2. 精通 CUDA 编程与 GPU 性能优化，有算子库（如 cuDNN、CUTLASS）开发或调优经验；
3. 熟练掌握 PyTorch/TensorFlow 框架机制，具备模型训练/推理部署实战经验；
4. 具备扎实的 C++/Python 能力，熟悉 Linux 开发环境及容器化工具链。

薪资：
面议

学历要求（及以上）：
本科

工作经验：
5-8年

年龄要求：
不限

其他要求：
加分项：
1. 有 LLM、视频生成等前沿模型优化经验，或参与过开源 AI 框架贡献；
2. 熟悉芯片架构（如 SIMT、内存层次结构），有 AI 芯片开发经验者优先；
3. 发表过高性能计算/体系结构相关顶会论文或专利。