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 Time深圳23h ago
-
Entry-level Full Time合肥1d ago
-
None Full Time上海1d ago
-
具身数据策略与质控工程师 CNY 25K-37KBenchmarking | Computer Vision | Data Quality | Data Quality Management | Data cleaningEntry-level Full Time北京1d ago
-
Mid-level Full Time北京1d ago
-
数据Infra工程师 (流式) CNY 180K-300KApache Spark | Automated Data Labeling | Data labeling | Data pipeline | Deep learningMid-level Full Time北京1d ago
-
数据Infra工程师 (具身) CNY 180K-300KApache Spark | Deep learning | Distributed Systems | Error Handling | FastAPIMid-level Full Time北京1d ago
-
Senior-level Full Time北京1d ago
-
Mid-level Full Time北京1d ago
-
Mid-level Full Time上海1d ago
-
Avp, Mgoc Tno Analytics, Gra CNY 300K-420KApplication development | Dash | Data Analysis | Data Visualization | DocumentationContinuous professional development | Flexible working | Inclusive work environmentExecutive-level Full TimeGuangzhou, Guangdong, China R1d ago
-
Application Engineer - Associate CNY 25K-37KAgent workflows | Application development | Autogen | Git | LLM Application DevelopmentEntry-level Part TimeShanghai, China1d ago
-
Lead Data Scientist CNY 435K-510KAI Agents | AWS | Azure | CI/CD | Cloud ComputingHealthcare | Paid volunteering days | Retirement planning | Wellbeing initiativesSenior-level Full TimeCHN-Beijing-Zhong Guan Cun No.1, China1d ago
-
C# | C++ | CUDA | Computer Vision | GazeboCareer development | Hands-on experienceEntry-level Full TimeChina, Shanghai1d ago
-
Junior AI Full-Stack Engineer CNY 96K-144KAPI Development | Agent workflows | Audit Logging | Authorization | CachingEntry-level Full TimeSJ China Shanghai1d ago
-
Entry-level Internship北京2d ago
-
大模型预训练数据算法工程师 CNY 216K-360KAblation Study | Data Deduplication | Data Engineering | Data Filtering | Data ParsingMid-level Full Time北京、上海2d ago
-
Entry-level Internship上海2d ago
-
Apache Flink | Apache Spark | Big Data | CI/CD | CassandraMid-level Full TimeShanghai2d ago
-
Analysis and Design | Big Data | Cassandra | Distillation | Distributed SystemsSenior-level Full TimeShanghai2d ago
-
Apache Flink | Apache Kafka | Apache Spark | Artificial Intelligence | CI/CDMid-level Full TimeShanghai2d ago
-
Senior-level Full Time北京2d ago
-
大模型算法工程师（多模态）-校招【27届】 CNY 240K-480KAB Testing | CLIP | Computer Vision | Contrastive Learning | Cross-modal retrievalSenior-level Full Time北京、杭州2d ago
-
Senior Software Engineer, Perception (Robotics) CNY 360K-510K3D Detection | Auto-labeling | BEV | Bayesian Filtering | C++Senior-level Full TimeShanghai, China2d ago
-
Enablement Lead - Data & AI Solution Enablement CNY 144K-240KAPI Integration | Agent automation | Artificial Intelligence | Data analytics | Low CodeSenior-level Full TimeShanghai, Shanghai, CN, 2012032d ago