VLA训练infra算法工程师 - XiaomiRobotics
Tasks
- Build high throughput data pipeline
- Design data sharding strategy
- Design distributed training for VLA models
- Develop training framework for stable efficient execution
- Implement distributed data parallel tensor parallel pipeline parallel mixture of experts training
- Implement operator fusion for performance
- Manage experiment tracking
- Optimize training with mixed precision
- Visualize training metrics
Perks/Benefits
- N/A
Skills/Tech-stack
BF16 | C++ | CUDA | Data Sharding | Data pipeline | DeepSpeed | Distributed Training | FP8 | FSDP | FlashAttention | Infiniband | Linux | Megatron-LM | Mixed Precision | NCCL | PyTorch | Python | RDMA | Triton
Education
N/A
Related jobs
-
Mid-level Full Time北京 R4h ago
-
Entry-level Full Time北京 R5h ago
-
Nlp / Llm 应用工程师 CNY 180K-360KAnthropic API | Chroma | Embedding | Genspark | Genspark RAG ArchitectureEntry-level Full Time北京 R5h ago
-
Entry-level Full Time北京 R5h ago
-
Ego-Centric 视觉感知算法工程师- Genie业务部 CNY 144K-240K3D Gaussian Splatting | 3D Pose Estimation | 3D point cloud | 6D Pose Estimation | Bundle adjustmentMid-level Full Time上海 R1d ago
-
2027届实习-算法数据开发工程师 CNY 96K-144KData Validation | Data Warehousing | Database fundamentals | Distributed Systems | ETLGlobal exposure | Hands-on projects | Learning opportunities | Technical mentorshipEntry-level Internship上海 R1d ago
-
Mid-level Full Time北京 R1d ago
-
Mid-level Full Time北京 R1d ago
-
优才-具身大模型训练框架工程师-觅蜂子公司 CNY 500K-500KC++ | CPU Optimization | CUDA | CUDA graph | Communication overlapMid-level Full Time上海 R5d ago
-
Diffusion Models | Flow matching | GPU Computing | Human Feedback | Multimodal LearningFully remote option | On-site flexibility | Open source collaborationMid-level Full TimeShanghai R5d ago
-
Generative AI - Graphics Engineer CNY 360K-600KBlender | C++ | Computer Graphics | DCC tool scripting | Data PipelinesFully remote option | On-site collaborationSenior-level Full TimeShanghai R5d ago
-
AWS | Access Control | Apache Flink | Apache Hive | Apache KafkaFully paid annual leave | Housing fund | No business trip | Social insurance | Work-life balanceSenior-level Full TimeHangzhou, Zhejiang Sheng, China R7d ago
-
【27届校招】数据运维工程师 CNY 180K-300KBig Data | Docker | Function Calling | Go | Hybrid CloudHands-on practice | MentorshipNone Full Time广州 R7d ago
-
具身智能算法工程师 CNY 25K-37KDeep learning | Distributed Training | GPU Training | IQL | Inference OptimizationInternshipEntry-level Internship北京 R7d ago
-
【27届校招】数据运维工程师 CNY 180K-300KAI Agent | Big Data | Containerization | Function Calling | GoMentorship | On-the-job trainingNone Full Time广州 R7d ago
-
Platform Operations Engineer CNY 306K-364KALCOA plus | API Management | Alerting | CI/CD | Canary ReleasesRemote workMid-level Full TimeShanghai, CN R8d ago
-
Behavior Cloning | C++ | Cloud processing | Control | DaggerEntry-level Internship北京、上海 R14d ago
-
Entry-level Full Time上海、深圳 R14d ago
-
Entry-level Internship上海、深圳 R14d ago
-
Deep Learning Compiler CI/Infrastructure Engineer CNY 160K-240KAI Agents | Agent workflows | Artifact management | Automated triage | AutomationGenerous benefits packageSenior-level Full TimeChina, Shanghai R16d ago
-
AI Engineer USD 100K-200KAPIs | Agentic Frameworks | Artificial Intelligence | Backtesting | Data integrationCollaborative work environment | Equity options | Innovation-driven cultureMid-level Full TimeShanghai, Shanghai, China R22d ago
-
Ai 院--多模态团队--多模态理解算法研究员-强化学习方向 CNY 240K-480KDPO | Data Preprocessing | Data cleaning | DeepSpeed | Distributed TrainingSenior-level Full Time北京 R1mo 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 R1mo ago
-
AI platforms | API Development | Artificial Intelligence | Cloud AI | Cloud AI PlatformsMid-level Full TimeRemote, China R1mo 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 R1mo ago