VLA训练infra算法工程师 - XiaomiRobotics
Tasks
- Build high throughput data pipeline
- Design data format and sharding strategy
- Design distributed training solutions
- Implement VLA model training framework
- Integrate operator fusion
- Manage experiment tracking and visualization
- Optimize mixed precision training
Perks/Benefits
- N/A
Skills/Tech-stack
BF16 | C++ | CUDA | DeepSpeed | Distributed Systems | FP8 | FSDP | FlashAttention | Infiniband | Linux | Megatron | Mixed Precision | NCCL | PyTorch | Python | RDMA | Triton
Education
N/A
Related jobs
-
Entry-level Full Time北京 R3h ago
-
Entry-level Full Time北京 R4h ago
-
具身智能算法工程师-模型 CNY 500K-500KActor-critic | Deep learning | Distributed Training | Implicit Q Learning | Inference accelerationMid-level Full Time北京 R4h ago
-
AI基础设施研发工程师(Sandbox / 容器化)-MiMo CNY 180K-420KAppArmor | Argo Workflows | CPU resource scheduling | Cgroup | ContainerdMid-level Full Time北京 R4h ago
-
Lead Embedded Software Engineer CNY 349K-437KARM | BLE | C# | C++ | Embedded LinuxHybrid work model | Remote-friendly | Work from homeSenior-level Full TimeSuzhou, China R4d ago
-
Mid-level Full Time上海、深圳 R10d ago
-
模型部署与推理优化工程师 CNY 180K-360KC++ | Edge inference | Inference Performance | Inference Performance Optimization | Model DistillationMid-level Full Time北京 R16d ago
-
Entry-level Internship上海 R17d ago
-
Mid-level Full Time北京 R1mo ago
-
AWS | Azure | JavaScript | NoSQL | Node.jsFast-paced environment | Remote workMid-level Full TimeHangzhou R1mo ago
-
AWS | Agile | Azure | Blockchain | CursorRemote workMid-level Full TimeShenzhen R1mo ago