猎头南楼
职位名称：AI算子优化工程师（GPU/CUDA方向）工作地点： 长沙 / 成都 / 北京薪资范围： 年薪30-60万（14薪）职位类型： 全职岗位职责｜What You‘ll Do1. 负责AI大模型（如Transformer类、多模态模型）在自研芯片上的性能极致优化，覆盖训练与推理全流程；2. 深入AI框架层（PyTorch/TensorFlow）与CUDA软件栈，进行算子开发、内核融合及内存/通信优化；3. 主导关键模型（如LLM、Diffusion）的分布式训练与推理性能调优，突破算力与通信瓶颈；4. 构建芯片级算子性能分析工具链，推动软硬件协同设计迭代。任职要求｜What We‘re Looking For1. 计算机/电子/数学等相关专业硕士及以上学历，3年以上AI底层开发经验；2. 精通CUDA编程与GPU性能优化，有算子库（如cuDNN、cutlass）开发或调优经验；3. 熟练掌握PyTorch/TensorFlow框架机制，具备模型训练/推理部署实战经验；4. 扎实的C++/Python能力，熟悉Linux开发环境及容器化工具链。加分项｜Nice to Have•有LLM、视频生成等前沿模型优化经验，或参与过开源AI框架贡献；•熟悉芯片架构（如SIMT、内存层次结构），有AI芯片开发经验者优先；•发表过高性能计算/体系结构相关顶会论文或专利。我们提供｜What We Offer•国产芯片头部平台： 参与国家重大战略项目，在自主芯片研发最前线实现技术价值；•有竞争力的薪酬： 年薪30-60万，14薪，五险一金实缴；•快速成长的通道： 汇报对象为技术总监，团队技术氛围浓厚，鼓励前沿探索与技术创新；•多城市灵活选择： 长沙、成都、北京三地任选，支持异地面试（有报销）。面试流程｜Interview Process一面（技术面，视频）→ 二面（HR面，视频）→ 三面（总监面，视频）
发布于 山西
1
评论
赞
未登录
友善发言
评论
加载中
