Machine Learning Engineer (Training Optimization）大模型训练优化工程师
Beijing, Beijing, China
CNY 192K-300K (estimate) Senior-level Full Time
Tasks
- Collaborate with research and modeling teams
- Debug and profile training workflows
- Design large scale machine learning systems for training
- Evaluate distributed training best practices
- Improve GPU utilization memory efficiency and communication
- Optimize distributed training performance
- Write custom GPU kernels
Perks/Benefits
- N/A
Skills/Tech-stack
CUDA | DeepSpeed | Diffusion Models | Distributed Training | FSDP | GPU Performance | GPU Performance Optimization | Gradient Checkpointing | JAX | Language Models | Large Language Models | Low Precision | Megatron-LM | Multimodal AI | NVIDIA Nemo | Performance optimization | PyTorch | Python | Triton | Zero
Education
N/A
Related jobs
-
Mid-level Full TimeChina - Hong Kong6h ago
-
LLM Agent Research Scientist 大模型智能体研究科学家 CNY 144K-240KAgent systems | Benchmarking | Cloud Computing | DeepSpeed | GPU ComputingInternship opportunity | Open to all experience levelsEntry-level Full TimeBeijing, Beijing, China8h ago
-
Entry-level Full TimeShanghai, China13h ago
-
Entry-level Internship广州18h ago
-
Mid-level Full Time深圳18h ago
-
Mid-level Full Time深圳18h ago
-
优才-具身大模型训练框架工程师-觅蜂子公司 CNY 500K-500KC++ | CPU Optimization | CUDA | CUDA graph | Communication overlapMid-level Full Time上海 R19h ago
-
Robotaxi VLA 大模型算法实习生 CNY 25K-37KAutonomous Driving | C++ | Case Development | Data cleaning | Data labelingEntry-level Internship广州19h ago
-
Applied AI Engineer CNY 300K-399KAutomation | Customer Relationship Management | Customer relationship | Evaluation | FintechEnglish working environment | Fast interview process | Remote workMid-level Full TimeChina19h ago
-
Senior Embedded Software Engineer CNY 360K-600KBSP | Bootloader | Buildroot | C Sharp | C plus plusSenior-level Full TimeShenzhen, Guangdong Province, China1d ago
-
Diffusion Models | Flow matching | GPU Computing | Human Feedback | Multimodal LearningFully remote option | On-site flexibility | Open source collaborationMid-level Full TimeShanghai R1d ago
-
Bash | Cloud platform | Data Processing | Docker | Google CloudAsynchronous culture | Competitive compensation | Friendly environment | Laid-back atmosphere | Remote-friendlyMid-level Full TimeShanghai, China1d ago
-
Mid-level Full Time北京1d ago
-
Mid-level Full Time北京1d ago
-
多模态数据算法工程师 CNY 216K-360KComputer Vision | Data Mining | Data Observability | Data cleaning | Deep learningEntry-level Full Time上海1d ago
-
具身智能数据算法工程师 CNY 144K-240K3D Gaussian Splatting | 3D Vision | Amazon S3 | Anomalous Data Detection | Apache ArrowMid-level Full Time北京1d ago
-
Mid-level Full Time深圳1d ago
-
Entry-level Full Time北京1d ago
-
多模态大模型算法工程师 CNY 240K-480KAblation Study | Agile | Computer Vision | Data Annotation | Data DeduplicationSenior-level Full Time北京1d ago
-
多模态大模型算法工程师 CNY 240K-480KComputer Vision | Data Augmentation | Data Deduplication | Data Mining | Data PreprocessingSenior-level Full Time深圳1d ago
-
多模态大模型算法工程师 CNY 240K-480KAgile | Computer Vision | Data Annotation | Data Deduplication | Data FilteringSenior-level Full Time上海1d ago
-
多模态大模型算法工程师 CNY 240K-480KAgile | Attribution Analysis | Automatic Labeling | Computer Vision | Data AblationSenior-level Full Time成都1d ago
-
Senior-level Full Time北京1d ago
-
Data Platform Engineering-Senior Principal, VP CNY 360K-600KAI Agent | AI Agent Development | AWS | Agent Development | Agent systemsEmployee networks | Flexible work/life support | Inclusive development opportunities | Paid volunteer daysSenior-level Full TimeHangzhou, China1d ago
-
Data Engineering - AI Platform Engineer CNY 300K-380KAWS | Access Control | Authentication | Azure | CI/CDEmployee networks | Flexible work/life support | Inclusive development opportunities | Paid volunteer daysMid-level Full TimeHangzhou, China1d ago