VLA训练infra算法工程师 - XiaomiRobotics
Tasks
- Build high throughput data pipeline
- Build metrics visualization
- Build training framework for stability and efficiency
- Design data formats and sharding strategy
- Design distributed training for VLA models
- Implement DP TP PP MoE parallelism
- Implement efficient data loading
- Manage large scale experiment tracking
- Optimize mixed precision training
- Optimize operator fusion with FlashAttention and Triton kernels
Perks/Benefits
- N/A
Skills/Tech-stack
BF16 | C++ | CPU architecture | CUDA | Data-parallel | DeepSpeed | Distributed Data Parallel | Distributed Training | Distributed data | FP8 | FSDP | FlashAttention | GPU Architecture | Infiniband | Kernel optimization | Linux | Megatron-LM | Memory Management | Mixed Precision | Mixture of Experts | Model Parallelism | NCCL | Network fundamentals | Pipeline parallelism | PyTorch | Python | RDMA | Tensor Parallelism | Triton
Education
N/A
Related jobs
-
Mid-level Full Time北京 R2d ago
-
Entry-level Full Time北京 R2d ago
-
Mid-level Full Time广州 R15d ago
-
Ai 院--多模态团队--多模态理解算法研究员-强化学习方向 CNY 240K-480KDPO | Data Preprocessing | Data cleaning | DeepSpeed | Distributed TrainingSenior-level Full Time北京 R19d ago
-
Lead Technical Support Engineer - AI / ML CNY 144K-240KAPI Integration | Agent Frameworks | C plus plus | Cause analysis | Cloud ComputingHybrid work model | Travel for customer workshops | Work from homeSenior-level Full TimeBeijing, China R19d ago
-
Nlp / Llm 应用工程师 CNY 180K-360KAgent evaluation | Anthropic API | Automated testing | Chroma | DeepEvalEntry-level Full Time北京 R21d ago
-
Mid-level Full Time北京 R21d ago
-
AI基础设施研发工程师（Sandbox / 容器化)-MiMo CNY 180K-420KAppArmor | Argo Workflows | CI/CD | CPU resource scheduling | CgroupMid-level Full Time北京 R21d ago
-
Entry-level Full Time北京、上海 R23d ago
-
Behavior Cloning | C++ | Cloud processing | Computer Vision | ControlEntry-level Internship北京、上海 R25d ago
-
Mid-level Full Time上海、深圳 R25d ago
-
AI platforms | API Development | Artificial Intelligence | Cloud AI | Cloud AI PlatformsMid-level Full TimeRemote, China R26d ago
-
AI ML Engineer CNY 280K-360KAWS | Azure | C++ | Cloud Computing | Computer VisionPerformance bonuses | Professional development opportunities | Remote workMid-level Full TimeShenzhen, Guangdong Province, China R29d ago
-
AI工程师-Agent Memory & RAG 方向（成都） CNY 240K-480KBERT | Chroma | Cross-Encoder | Embedding Models | FaissSenior-level Full Time成都 R30d ago
-
AI工程师-Agent Memory & RAG 方向（武汉） CNY 240K-480KAlgorithms | BERT | Chroma | Cross-Encoder | Data StructuresSenior-level Full Time武汉 R30d ago
-
AI工程师-Agent Memory & RAG 方向（北京） CNY 240K-480KBERT | Chroma | Cross-Encoder | Embedding Models | FaissSenior-level Full Time北京 R30d ago
-
AWS | Agent Orchestration | Agent systems | Azure | DockerMid-level Full TimeShenzhen, Guangdong, China R1mo ago
-
AVP, AI Solution Lead CNY 360K-600KCloud Computing | DataOps | DevOps | Flutter | Generative AIContinuous professional development | Flexible workingSenior-level Full TimeGuangzhou, Guangdong, China R1mo ago
-
AWQ | AWS | Accelerate | Azure | BatchingMid-level Full TimeShenzhen, Guangdong, China R1mo ago
-
Generative AI - ML System Engineering CNY 360K-600KC++ | CUDA | Compilation | Data pipeline | Diffusion ModelsFully remote option | On-site work flexibilitySenior-level Full TimeShanghai R1mo ago
-
Lead Embedded Software Engineer CNY 349K-437KARM | BLE | C# | C++ | Embedded LinuxHybrid work model | Remote-friendly | Work from homeSenior-level Full TimeSuzhou, China R1mo ago
-
Mid-level Full Time上海、深圳 R1mo ago
-
模型部署与推理优化工程师 CNY 180K-360KC++ | Edge inference | Inference Performance | Inference Performance Optimization | Model DistillationMid-level Full Time北京 R1mo ago
-
Entry-level Internship上海 R1mo ago