顶尖应届-大模型AI算力研发工程师
北京
硕士及以上
计算机类
使用简历深度优化功能,快速提升简历质量
职位介绍
【课题名称】
GPU 国产卡适配研究
【课题内容】
1、研究国产卡技术栈并完成适配开发,支撑大模型推理等场景接入使用
2、研究主流 GPU卡特性并掌握 CUDA 生态等,支撑大模型训练和推理场景的性能优化和问题定位
职位描述
1、GPU 国产卡的适配开发;
2、大模型训推场景 GPU 相关性能优化和问题定位。
职位要求
1、计算机相关专业,硕士及以上学历;
2、熟悉主流 GPU 训练和推理芯片,熟悉 CUDA 生态,有 CUDA 开发和性能调优经验优先;
3、熟悉容器及 Kubernetes 技术,具备基于 Kubernetes 的 GPU 算力管理经验者优先;
4、熟练 Go/Python/C++至少一门语言,有 PyTorch/Volcano/Megatron/SGLang 框架熟悉者优先。

