Xiaomi Technology

AI算力工程师

Xiaomi Technology  •  Wuhan, CN / Shanghai, CN / Nanjing, CN / Beijing, CN (Onsite)  •  1 day ago
Apply
AI can make mistakes so check important info. Chat history is never stored.

Job Description

AI算力工程师
武汉、北京、南京、上海
校招
正式
软件研发类
2027届校园招聘计划
职位描述
1、GPU 核心能力建设:持续探索前沿 GPU 芯片的技术特性,包括国产卡适配、混合精度、模型量化、显存优化等技术,将芯片能力转化为业务竞争力,有力支撑集团 AI 战略落地;
2、万卡集群调度与管理:基于 Kubernetes 建设和运营万卡级 GPU 集群,负责 GPU 适配、拓扑感知调度、Gang 调度、GPU 故障自愈、全链路可观测等核心能力,支撑小米自研大模型 MiMo 的训练和推理等重点业务场景;
3、GPU 池化和利用率提升:统一纳管小米 GPU 资源,推进 GPU 调度优化、资源池化、GPU 虚拟化、训推混部等关键技术研发,持续提升全集团 GPU 资源利用率,实现技术降本增效。
职位要求
1、熟悉主流 GPU 训练和推理芯片,熟悉 CUDA 生态,有大模型训练和推理优化经验优先;
2、熟悉 Kubernetes 和 GPU 算力管理,有万卡集群建设和 GPU 利用率提升经验优先;
3、熟练 Go/Python/C++至少一门语言,有 PyTorch/Volcano/Megatron/SGLang 框架熟悉者优先
计算机相关专业,本科及以上学历。
投递
Xiaomi Technology

About Xiaomi Technology

Xiaomi Corporation was founded in April 2010 and listed on the Main Board of the Hong Kong Stock Exchange on July 9, 2018 (1810.HK). Xiaomi is a consumer electronics and smart manufacturing company with smartphones and smart hardware connected by an IoT platform at its core.

Embracing our vision of “Make friends with users and be the coolest company in the users’ hearts”, Xiaomi continuously pursues innovations, high-quality user experience and operational efficiency. The company relentlessly builds amazing products with honest prices to let everyone in the world enjoy a better life through innovative technology.

Xiaomi is one of the world's leading smartphone companies. The company has also established the world’s leading consumer AIoT (AI+IoT) platform,reached 558 million smart devices connected to its platform (excluding smartphones,laptops and tablets) as of September 30 2022. Xiaomi products are present in more than 100 countries and regions around the world. In August 2022, Xiaomi was included in the Fortune Global 500 list for the fourth year in a row, ranking 266th. The company is the fastest-rising Chinese technology conglomerate during the four-year period.

Xiaomi is a constituent of the Hang Seng Index, Hang Seng China Enterprises Index, Hang Seng TECH Index and Hang Seng China 50 Index.

Industry
IT & Software
Company Size
10,000+ employees
Headquarters
Beijing, CN
Year Founded
2010
Social Media