Xiaomi Technology

自动驾驶 - 大模型训练框架优化专家

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

Job Description

自动驾驶 - 大模型训练框架优化专家北京社招全职职位 ID:A43941职位描述1. 负责分析和优化大模型的分布式训练链路,提升训练性能和效率;
2. 对大模型训练优化技术展开研究,并落地到自动驾驶;
3. 负责推进训练框架和AI平台的结合,建立先进的训练调度机制、集群算力利用率评估机制等。职位要求1. 本科及以上学历,扎实的编程基础(Python/C++)与良好的工程习惯
2. 精通深度学习框架底层原理(PyTorch 核心机制如Torch.Compile, Autograd、TorchScript, XLA),
3. 熟悉分布式训练框架(Horovod、Megatron-LM、DeepSpeed 等)的实现逻辑.
4. 精通 CUDA 编程及 GPU 架构,能独立开发高性能自定义算子(cuda,CUB,Thrust,cutile,TileLang,Triton)
4. 有深度学习编译技术栈(TVM,MLIR等),、AI开源社区贡献等任一经验者优先
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