Xiaomi Technology

训练平台性能优化工程师实习生

Xiaomi Technology  •  Remote  •  22 days ago
Apply
AI can make mistakes so check important info. Chat history is never stored.

Job Description

训练平台性能优化工程师实习生北京校招实习软件研发类职位描述深度优化训练流程
主导模型训练全链路性能分析与优化,设计GPU资源弹性调度策略
开发自动化训练加速工具链,构建可扩展的云端训练框架
研发混合精度训练、梯度压缩等前沿技术,突破训练吞吐瓶颈
构建训练优化体系
制定标准化训练效能评估体系,建立成本-效率量化模型
设计可复用的训练加速组件库,沉淀最佳实践方法论
开发训练过程性能分析平台,实现性能问题智能诊断
赋能业务研发
优化多任务资源调度策略,提升GPU集群整体利用率
为算法团队提供训练加速解决方案,缩短模型迭代周期职位要求1.精通深度学习训练加速技术,熟悉分布式训练框架设计
2.掌握CUDA编程及GPU性能分析工具(nsys/torch profiler)
3.熟练使用PyTorch框架,具备训练流程优化实战经验
4.熟悉常见模型压缩技术(量化/剪枝/蒸馏)及落地应用
5.具备大规模集群资源调度系统开发经验者优先
【技术加分项】
1.有mmcv//deepspeed/megatron/ray等训练框架开发经验
2.算子优化经验,triton/cuda等开发经验.
3.熟悉MPI/NCCL等分布式通信协议
4.在MLSys/ICLR等顶会发表过训练优化相关论文 投递
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
Website
mi.com
Social Media