Xiaomi Technology

VLA训练infra算法工程师 - XiaomiRobotics

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

Job Description

VLA训练infra算法工程师 - XiaomiRobotics北京社招全职职位 ID:A243642职位描述1. 基于 PyTorch 生态(FSDP / DeepSpeed / Megatron 等)设计并实现 VLA 模型的分布式训练方案(DP / TP / PP / MoE),构建稳定高效的训练框架
2. 推动混合精度(BF16 / FP8)与算子融合(FlashAttention / Triton kernel)
3. 构建高吞吐数据pipeline,设计数据格式与 shard 策略,实现高效的数据加载
4. 支持大规模实验追踪、管理、指标可视化职位要求1. 扎实的 Python / C++ 基础,熟悉 Linux 系统
2. 深入理解 PyTorch 分布式训练, 熟悉至少一种分布式训练优化框架(FSDP / DeepSpeed / Megatron)
3. 深入理解 GPU / CPU / 内存 / 网络 基本原理
4. 有大规模训练经验
加分项
1. 有 多模态 / VLA / 机器人 / 自动驾驶 大规模训练经验
2. 熟悉 NCCL / RDMA / InfiniBand 调优
3. 有自定义 CUDA / Triton 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