Xiaomi Technology

顶尖实习-大模型高性能算子研发工程师-AI基建

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

Job Description

顶尖实习-大模型高性能算子研发工程师-AI基建

北京

校招

实习

软件研发类

【使用中】顶尖实习计划

职位描述

  • 基于 CUDA / CUTLASS 以及 Triton / TileLang 等算子开发工具,针对不同模型结构与硬件架构进行算子设计、实现、融合(kernel fusion)与调优,提升计算访存效率与硬件利用率(MFU) - 使用 nsys / ncu 等性能分析工具定位计算、访存、通信瓶颈,开展 roofline 分析与 kernel 级优化,持续提升推理吞吐、降低时延。 - 跟踪业界前沿(新模型结构、新硬件特性、开源算子库等),参与算子库开发和建设、性能基准测试与正确性验证

职位要求

  • 硕士及以上学历,计算机、软件工程、数学等相关专业,具备扎实计算机理论基础和丰富编程经验,在深度学习领域有深入研究与实践。 - 精通 Python,熟练掌握至少一种深度学习框架(如 PyTorch),理解深度学习和transformer系列算法,有丰富模型开发与调优经验,能独立搭建复杂模型并优化性能。 - 熟悉主流的并行编程以及性能优化技术,熟悉CUDA和TensorCore编程,有相关高性能优化经验(nsys/ncu)优先; - 熟悉flashattention,flashinfer, deepgemm等算子库, 有使用triton/tilelang等做高性能算子开发经验优先 - 有良好团队协作与沟通能力,能与不同背景成员有效沟通协作,责任心强、敬业热情,能承受工作压力,保证项目按时高质量交付。

投递

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