招聘信息分享 招聘信息分享 ← 返回岗位列表

顶尖应届-大模型AI算力研发工程师-AI 基建

小米📍北京集团技术委员会
发布时间:

岗位职责

1、GPU 国产卡的适配开发; 2、大模型训推场景 GPU 相关性能优化和问题定位。

任职要求

1、计算机相关专业,硕士及以上学历; 2、熟悉主流 GPU 训练和推理芯片,熟悉 CUDA 生态,有 CUDA 开发和性能调优经验优先; 3、熟悉容器及 Kubernetes 技术,具备基于 Kubernetes 的 GPU 算力管理经验者优先; 4、熟练 Go/Python/C++至少一门语言,有 PyTorch/Volcano/Megatron/SGLang 框架熟悉者优先。
前往官网申请 →