小米 校园招聘
顶尖应届-大模型AI算力研发工程师-AI 基建
北京 ·经验不限·学历不限
北京
薪资面议
前往 小米 官网投递
职位描述
1、GPU 国产卡的适配开发;
2、大模型训推场景 GPU 相关性能优化和问题定位。
【任职要求】
1、计算机相关专业,硕士及以上学历;
2、熟悉主流 GPU 训练和推理芯片,熟悉 CUDA 生态,有 CUDA 开发和性能调优经验优先;
3、熟悉容器及 Kubernetes 技术,具备基于 Kubernetes 的 GPU 算力管理经验者优先;
4、熟练 Go/Python/C++至少一门语言,有 PyTorch/Volcano/Megatron/SGLang 框架熟悉者优先。
官网发布:2026-07-29 · 最后确认在招:2026-09-03 23:17:24 · 来源平台:xiaomi