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
-
Mid-level Full Time武汉、北京1d ago
-
Senior-level Full Time武汉、北京1d ago
-
Senior-level Full Time北京、广州、上海1d ago
-
Data Processing | Deep learning | Distributed Training | Generative Models | LLMFamily leave | Free food and snacks | Health care plan | Life insurance | Long-term disabilitySenior-level Full Time费利蒙1d ago
-
(Senior) Software Engineer, Deep Learning USD 140K-280KAlgorithms | C++ | CUDA | Code optimization | Computational GeometryFamily leave | Free food and snacks | Health care plan | Life insurance | Long-term disabilitySenior-level Full Time费利蒙1d ago
-
Senior Software Engineer - ASE Data Services CNY 160K-240KApache Kafka | Apache Solr | Apache Zookeeper | C# | Capacity PlanningCollaborative work environment | Technical mentorship | Work-life balanceSenior-level Full TimeShanghai1d ago
-
Senior Deep Learning Engineer, 4D Foundation Model CNY 360K-600K3D Geometry | CUDA | Camera Calibration | Computer Vision | Coordinate systemsSenior-level Full TimeChina, Shanghai2d ago
-
Ai全栈工程师 CNY 144K-240KAmazon Web Services | CI/CD | Docker | Express | FastAPIAI tools license | Flexible work hours | Remote work optionSenior-level Full Time北京、南京、苏州2d ago
-
Lead Engineer, Machine Learning CNY 360K-500KCost Optimization | Data Pipelines | Fine Tuning | GPU Optimization | Inference SystemsSenior-level Full TimeChina R2d ago
-
Principal Software Engineer - Apple Data Platform CNY 435K-510KAWS | Airflow | Alibaba Cloud | Automation | C++Senior-level Full TimeShanghai2d ago
-
AWS | Agent Orchestration | Agent systems | Azure | Cloud platformMid-level Full TimeShenzhen, Guangdong, China R2d ago
-
AWS | Access Management | Agentic Workflows | Auditability | AzureMid-level Full TimeShenzhen, GD, CN, 5180103d ago
-
APIs | AWS | Accuracy Optimization | Agent Orchestration | Agent systemsMid-level Full TimeShenzhen, GD, CN, 5180103d ago
-
Deep Learning Solution Architect - Agentic Performance CNY 360K-600KAI Agents | Agentic Inference | Cluster management | Distributed Training | Distributed inferenceSenior-level Full TimeChina, Beijing3d ago
-
Senior-level Full TimeChina, Shanghai3d ago
-
Mid-level Full TimeChina, Shanghai3d ago
-
AI Infra工程师-算法研发方向 CNY 180K-300KC plus plus | CI/CD | CUDA | Data parallelism | Deep learningEngineering Training | MentorshipMid-level Full Time北京、上海3d ago
-
AI Software Development Eng. CNY 74K-100KCUDA | Data parallelism | Debugging | DeepSpeed | Distributed TrainingEntry-level Full TimeCN,Shanghai WFOE3d ago
-
大模型算法工程师（数据方向） CNY 25K-37KAblation Testing | Apache Spark | Benchmarking | Classifier Based Filtering | Cloud ComputingEntry-level Full TimeHaidian, Beijing, China4d ago
-
Entry-level Full Time合肥4d ago
-
Apache Flink | Apache Iceberg | Apache Spark | CI/CD | DockerMid-level Full TimeShanghai R4d ago
-
Apache Flink | Apache Iceberg | Apache Spark | CI/CD | DatabasesMid-level Full TimeShanghai R4d ago
-
None Full Time上海4d ago
-
None Full Time北京、上海4d ago
-
Mid-level Full Time北京、上海4d ago