直招.cv

← 返回职位列表

小米 社会招聘

新业务部-大模型部署优化工程师

北京 ·经验不限·学历不限

北京·上海·西安

职位描述

1. 参与Android 平台的大模型(aigc + llm)开发和部署; 2. 负责大模型算子在NPU/GPU/CPU部署及高性能算子实现; 3. 负责大模型异构部署的算子性能调优及优化工作; 4. 分析算子/网络的性能瓶颈,提出性能优化策略,达成性能目标。 【任职要求】 1. 本科及以上学历,计算机科学、电子工程或自动化等相关专业; 2. 熟悉并行/异构计算体系结构,在TPU/NPU/GPU一种平台有3年以上的开发和优化经验; 3. 熟悉计算机系统体系架构,软件性能优化加速; 4. 熟练使用至少一种主流Al训练框架,如TensorFlow、PyTorch、Caffe等; 5.具备业界常用高性能库(TensorRT/cuDNN/CUDA/OPENCV)开发和性能优化经验者优先; 6. 良好的团队沟通能力、技术攻关能力,责任心强。

官网发布:2025-08-03 · 最后确认在招:2026-09-03 23:18:29 · 来源平台:xiaomi