Xiaomi Technology

大模型训练/推理框架负责人

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

Job Description

大模型训练/推理框架负责人北京社招全职职位 ID:A245754职位描述1. 负责大语言模型训练框架开发和优化,包括不限于性能分析,高性能算子,计算和通信优化,可观测性等
2. 负责大语言模型推理框架开发和优化,包括不限于PD分离,专家并行,高性能算子等
3. 负责训练和推理框架团队管理和规划,支持上层大语言和多模态模型的训练和推理职位要求1. 计算机/电子/数学等相关专业硕士及以上学历, 8年以上计算机相关工作经验,3年以上团队管理经验
2. 编程能力 :精通C/C++,熟悉Python,具备扎实的数据结构与算法基础
3. GPU与CUDA :熟悉NVIDIA GPU架构及编程模型,掌握CUDA核函数优化、显存管理、多流并发等技术,有实际性能调优经验
4. 熟悉PyTorch、Megatron、vLLM/SGLang等深度学习训练和推理框架, 有分布式系统开发经验
5. 熟悉常用大语言模型,多模型以及生成模型等算法和架构,参与过大模型训练/推理项目 投递
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