VLA训练infra算法工程师 - XiaomiRobotics
Tasks
- Build training framework for stability and performance
- Design data format and sharding strategy
- Design distributed training solutions for VLA models
- Develop high throughput data pipeline
- Implement DP TP PP MoE training strategies
- Implement operator fusion for faster training
- Manage large scale experiment tracking and metrics visualization
- Optimize mixed precision training
Perks/Benefits
- N/A
Skills/Tech-stack
BF16 | C++ | CUDA | Data pipeline | DeepSpeed | Distributed Training | FP8 | FSDP | FlashAttention | Infiniband | Linux | Megatron | Mixed Precision | NCCL | PyTorch | Python | RDMA | Triton
Education
N/A
Roles
Related jobs
-
None Full Time上海3h ago
-
Entry-level Full Time北京4h ago
-
机器人VLA算法研究员 - XiaomiRobotics CNY 500K-500KAction Generation | Data Modeling | Deep learning | Diffusion Models | Machine LearningMid-level Full Time北京4h ago
-
具身世界模型训练INFRA工程师 - XiaomiRobotics CNY 180K-360KAPIs | Deep learning | Distributed Training | Fault Tolerance | Language ModelsMid-level Full Time北京4h ago
-
具身世界模型推理INFRA工程师 - XiaomiRobotics CNY 144K-240KCFG Parallelism | Diffusion Models | Expert parallelism | FP8 | Model QuantizationSenior-level Full Time北京4h ago
-
大数据开发工程师 CNY 180K-300KAnomaly Detection | Apache Flink | Apache Spark | Data Governance | Data LakeMid-level Full Time北京4h ago
-
Entry-level Full Time InternshipBeijing4h ago
-
Mid-level Full Time北京5h ago
-
Mid-level Full Time北京5h ago
-
Senior-level Full Time北京、上海5h ago
-
Senior-level Full Time上海、北京5h ago
-
Senior-level Full Time北京、上海5h ago
-
自动驾驶数据挖掘算法开发实习生(Llm/Vlm方向)-2027届 CNY 25K-37KAccuracy | Algorithm Reproducibility | Computer Vision | Coverage | Data PreprocessingEntry-level Internship武汉5h ago
-
技术中心 - 数据开发工程师 CNY 180K-300KData Lineage | Data Modeling | Data Monitoring | Data Observability | Data QualityMid-level Full Time广州5h ago
-
技术中心 - 数据开发工程师 (游戏业务方向) CNY 120K-180KData Modeling | Data Quality | Data Warehouse | Data pipeline | FlinkMid-level Full Time广州5h ago
-
Entry-level Full Time上海5h ago
-
Alerting | Benchmarking | Data Pipelines | Distributed Systems | GPU ComputingMid-level Full TimeChina R18h ago
-
AI Lead Engineer CNY 120K-180KAWS | Ansible | Azure | CI/CD | DockerDevelopment courses | Employee assistance program | Flexible working | Learning opportunities | Medical and life insuranceSenior-level Full TimeTianjin, CN18h ago
-
Mid-level Full TimeChina R18h ago
-
Mid-level Full TimeShanghai, China23h ago
-
Senior-level Full TimeWuxi, Jiangsu, China23h ago
-
Embedded Cybersecurity Engineer(Platform Security) CNY 74K-100KAccess Control | Anti-rollback | AppArmor | Authenticated Service Port | CVE AnalysisEntry-level Full TimeWuxi, Jiangsu, China23h ago
-
Senior Data Platform & Annotation Engineer CNY 360K-600KArtificial Intelligence | Big Data | C++ | CI/CD | Camera image processingSenior-level Full Time5-8F TOWER C, 788 JINZHONG ROAD, …23h ago
-
Entry-level Full Time上海1d ago
-
2026校招-Ai院-开源模型适配 & 技术布道 CNY 180K-264KGitHub | Hugging Face | Hugging Face Transformers | Inference Optimization | Model ArchitectureCareer development | Open source community impact | Team collaborationNone Full Time北京1d ago