职责描述
职责描述:
【岗位职责】
· 参与推荐GR大模型的NVIDIA GPU算子开发
· 进行算子性能分析与优化,提升训练和推理效率
· 协助调研前沿算子优化技术并落地验证
任职要求:
【能力】
[必备]
· C++或CUDA编程基础,了解GPU并行计算原理
· PyTorch框架使用经验
[加分]
· Attention或MoE模型结构理解
· NVIDIA GPU算子优化实践经验
· 高性能计算或编译优化相关项目经历
【特质】
[必备]
· 对底层系统优化有热情,愿意深入钻研
· 学习能力强,能快速掌握新技术栈