AI Infrastructure Lead
Tasks
- Build distributed training infrastructure
- Build experiment tracking and model registry
- Collaborate with research teams for model and reinforcement learning training
- Collaborate with robotics teams for model inference deployment
- Design machine learning infrastructure architecture
- Develop distributed training systems for large models
- Implement training pipeline for dataset preprocessing training evaluation deployment
- Improve training stability
- Lead ML infrastructure team
- Manage GPU cluster training systems
- Optimize GPU utilization and communication efficiency
Perks/Benefits
- N/A
Skills/Tech-stack
CUDA | DeepSpeed | Experiment tracking | FSDP | Horovod | JAX | Kubernetes | Megatron | Memory Optimization | Mixed Precision | Model Registry | NCCL | PyTorch | Ray | Slurm | TensorFlow
Education
Regions
Countries
States
Related jobs
-
Senior-level Full Time上海、北京21h ago
-
Entry-level Internship上海21h ago
-
Mid-level Full Time上海21h ago
-
Mid-level Full Time上海、深圳 R21h ago
-
Mid-level Full Time上海21h ago
-
Mid-level Full Time上海21h ago
-
Mid-level Full Time上海、北京21h ago
-
3D视觉算法工程师 CNY 180K-300K3D Pose Estimation | 3D Reconstruction | 6D Pose | 6D Pose Estimation | Audio ProcessingMid-level Full Time深圳21h ago
-
Senior-level Full Time上海 R21h ago
-
Mid-level Full Time上海 R21h ago
-
Senior-level Full Time上海 R21h ago
-
loco-manipulation(高动态物体交互方向) CNY 180K-360KBody Dynamics | C++ | Contact dynamics | Curriculum learning | Domain RandomizationMid-level Full Time上海21h ago
-
Data Infra高级研发工程师(AI与具身智能数据平台方向) CNY 144K-240KAsynchronous processing | Automated testing | CI/CD | CRD | Cloud NativeSenior-level Full Time上海 R21h ago
-
Mid-level Full Time北京、上海21h ago
-
Ego-Centric视觉感知算法工程师 CNY 180K-300K2D Segmentation | 3D Gaussian Splatting | 3D Hand Pose | 3D Reconstruction | 6D PoseMid-level Full Time上海、北京 R21h ago
-
Senior-level Full Time上海22h ago
-
数据算法开发工程师/专家 CNY 240K-480K2D Computer Vision | 3D Computer Vision | ASR | Ablation Study | Active LearningSenior-level Full Time上海 R22h ago
-
多模智能交互算法专家 CNY 144K-240K3D Vision | Agentic OS | Autonomous Planning | DeepSpeed | Distributed TrainingSenior-level Full Time上海、北京22h ago
-
Mid-level Full Time上海22h ago
-
多模态数据评测工程师 CNY 36K-42KAnnotation | Audio Video Data Alignment | Audio/Video | Automation | BenchmarkingMid-level Full Time上海22h ago
-
资深机器视觉专家 CNY 180K-300KComputer Vision | Deep learning | Docker | Image Segmentation | Image classificationMid-level Full Time北京、上海23h ago
-
Software Engineer - Generative AI & ML CNY 144K-240KAWS | Agent Orchestration | CI/CD | Cloud platform | Data PreprocessingSenior-level Full TimeShanghai1d ago
-
Site Reliability Engineer, Ai & Data Platforms CNY 144K-240KAnsible | Apache Airflow | Apache Spark | Argo CD | BashOn-call rotationSenior-level Full TimeShanghai1d ago
-
Consultant Specialist (GenAI Agent Platform) CNY 300K-420KAWS | Access Control | Agent Orchestration | Audit Logging | AzureEntry-level Full TimeGuangzhou, Guangdong, China R1d ago
-
Senior-level Full TimeShanghai1d ago