Xiaomi Technology

模型优化专家/工程师

Xiaomi Technology  •  Onsite  •  2 hours ago
Apply
AI can make mistakes so check important info. Chat history is never stored.

Job Description

模型优化专家/工程师
北京、上海
社招
全职
职位 ID:A214755
职位描述
1. 负责AI模型的网络级压缩与优化,涵盖量化(QAT/PTQ)、投机采样(Speculative Decoding)、Token压缩、知识蒸馏等技术;
2. 针对大小模型设计并实现差异化量化方案,如小模型的KL散度、Percentile、AdaRound等,大模型的SmoothQuant、AWQ等;
3. 参与或主导模型网络结构的优化与重构,将现有网络的算子、Block替换为高效、硬件友好的架构;
4. 参与模型训练与微调,能够复现、分析并改进前沿模型结构;
5. 与硬件团队紧密合作,提供模型优化视角的硬件设计建议,推动下一代芯片对新型算子与网络结构的支持。
职位要求
1. 计算机科学、人工智能、数学等相关专业硕士及以上学历,2年以上模型优化或训练经验;
2. 深入理解模型量化技术(PTQ/QAT),至少熟悉一种小模型或大模型量化方案,并有实际部署经验;
3. 熟悉大模型推理优化技术,如投机采样、Token剪枝、稀疏化等;
4. 具备较强的模型训练与复现能力,能够阅读论文并快速实现网络结构改进;
5. 了解芯片计算特性,有与硬件团队协作进行模型-芯片协同设计经验者优先;
6. 代码能力强,熟练使用PyTorch/TensorFlow等框架,熟悉底层算子开发(如CUDA kernel)者更佳。
投递
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