Machine Learning Engineer (Training Optimization)
Beijing, Beijing, China
CNY 240K-480K (estimate) Senior-level Full Time
Tasks
- Align system design with model requirements
- Debug and profile training workflows
- Design distributed training systems
- Develop custom GPU kernels
- Evaluate distributed training best practices
- Fine tune training workflows for scalability
- Implement large scale model training pipelines
- Improve memory efficiency
- Optimize GPU utilization
- Reduce communication overhead
Perks/Benefits
- N/A
Skills/Tech-stack
CUDA | Data Types | DeepSpeed | Diffusion Models | Distributed Training | FSDP | GPU Optimization | Gradient Checkpointing | JAX | LLMs | Language Models | Large Language Models | Low Precision Data Types | Megatron-LM | Memory Optimization | Multimodal AI | NVIDIA Nemo | PyTorch | Python | Triton | Zero
Education
N/A
Related jobs
-
Entry-level Internship上海17h ago
-
Entry-level InternshipShanghai1d ago
-
Mid-level Full Time上海1d ago
-
Senior-level Full Time上海1d ago
-
【算法】多模态/大模型算法专家（上海） CNY 240K-480KAgent Frameworks | C++ | Computer Vision | Language Processing | LinuxSenior-level Full Time上海1d ago
-
【算法】计算机视觉算法专家（上海） CNY 240K-480KAnomaly Detection | C++ | Computer Vision | Deep learning | Fine-grained classificationSenior-level Full Time上海1d ago
-
Applied AI Engineer CNY 330K-500KCost Optimization | Fine Tuning | JAX | LLM Inference | Language ModelsSenior-level Full TimeChina1d ago
-
Sr. Business Intelligence Engineer , CNGS NBS (New Business & New Seller) Amazon Business CNY 180K-500KBenchmarking | Data Analysis | Data Mapping | Data Modeling | Data NormalizationSenior-level Full TimeShanghai, CHN2d ago
-
Mid-level Full Time5-8F TOWER C, 788 JINZHONG ROAD, …2d ago
-
AI Engineer CNY 240K-360KAI workflow | AI workflow design | Agent systems | Computer Vision | Data AnnotationOccasional travel | Office environmentSenior-level Full TimeChina - Suzhou, Jiangsu - 297 …2d ago
-
Lead Hardware & Software Test Engineer CNY 300K-480KAI machine learning | Agile | Automation frameworks | Automation libraries | Cloud TestingSenior-level Full TimeChina - Shenzhen Office2d ago
-
Entry-level Full Time北京、上海2d ago
-
Mid-level Full Time北京 R2d ago
-
数据开发工程师（AI Agent方向） CNY 216K-360KAPI Development | AST | Data Dictionary | Data Governance | Data ModelingMid-level Full TimeBeijing2d ago
-
Data Analysist / Digital Consultant CNY 304K-399KAPI Integration | AWS Data | AWS Data Lake | Amazon Web Services | Apache SparkFlexible working hours | Inclusive work environment | Work-life balanceMid-level Full TimeShanghai, Asia, China2d ago
-
Principal Perception Engineer CNY 380K-540K3D Geometry | BEV Transformation | C++ | CNN | Camera CalibrationSenior-level Full Time5-8F TOWER C, 788 JINZHONG ROAD, …3d ago
-
Senior Perception Engineer CNY 360K-540K3D Geometry | BEV Transformation | C++ | Camera Calibration | Computer VisionSenior-level Full Time5-8F TOWER C, 788 JINZHONG ROAD, …3d ago
-
Senior Data Engineer, Off CNY 260K-366KAWS | Apache Iceberg | Apache Spark | Batch Processing | CI/CDEmployee networks | Flexible work/life support | Inclusive development opportunities | Paid volunteer daysSenior-level Full TimeHangzhou, China3d ago
-
AWS | Apache Iceberg | Apache Spark | Automated testing | AzureEmployee networks | Flexible work/life support | Inclusive development opportunities | Paid volunteer daysSenior-level Full TimeHangzhou, China3d ago
-
Senior Data Engineer, Off CNY 260K-366KAPIs | AWS | Apache Iceberg | Apache Spark | BIEmployee networks | Flexible work/life support | Inclusive development opportunities | Paid volunteer daysSenior-level Full TimeHangzhou, China3d ago
-
AWS | Access Control | Apache Iceberg | Authentication | AzureEmployee networks | Flexible work/life support | Inclusive development opportunities | Paid volunteer daysSenior-level Full TimeHangzhou, China3d ago
-
Senior-level Full TimeChina, Shanghai3d ago
-
AI Model Inference | AI model | Agent Framework | C++ | Closed LoopEntry-level Full Time InternshipChina, Shenzhen3d ago
-
Automated testing | C# | C++ | Datalink communication | DebuggingEmployee assistance programs | Flexible spending accounts | Health Lifestyle Programs | Health savings account | Life insuranceMid-level Full TimeWuxi, Jiangsu, China3d ago
-
Senior-level Full Time北京、苏州3d ago