Research MLE (Training Optimization）大模型训练优化工程师
Beijing, Beijing, China
CNY 144K-240K (estimate) Senior-level Full Time
Tasks
- Collaborate with research and modeling teams
- Debug profile and fine tune training workflows
- Design distributed training systems
- Develop low level optimization for training
- Evaluate and apply distributed training best practices
- Implement and optimize large scale machine learning systems
- Improve GPU utilization
- Optimize memory efficiency
- Reduce communication overhead
- Write custom CUDA or Triton kernels
Perks/Benefits
- N/A
Skills/Tech-stack
C++ | CUDA | DeepSpeed | Distributed Training | FSDP | GPU Kernel | GPU kernel optimization | Gradient Checkpointing | JAX | Kernel optimization | Low Precision | Megatron-LM | NVIDIA Nemo | PyTorch | Python | Rust | Triton | Zero
Education
N/A
Related jobs
-
Mid-level Full Time深圳23h ago
-
Entry-level Full Time合肥1d ago
-
None Full Time上海1d ago
-
数据科学家-2027届 CNY 180K-360KAgentic Workflows | Clustering | Data Governance | Hypothesis Testing | Language ModelsNone 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
-
Artificial Intelligence | Data Mining | Machine Learning | Predictive Modeling | PythonMid-level Full TimeCN-OCG International Center, Cheng Du, China1d ago
-
A/B | A/B Testing | B testing | Data Modeling | Entity ResolutionWorkplace accommodations for hiring processMid-level Full TimeShanghai, CHN1d 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
-
Data Mining | Data Modeling | Language Processing | Machine Learning | Natural LanguageMid-level Full TimeCN-OCG International Center, Cheng Du, China1d ago
-
Senior Data Scientist CNY 480K-540KAutoml | Data Visualization | Deep learning | H2O | Machine LearningSenior-level Full TimePudong, Shanghai, 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
-
Apache Flink | Apache Kafka | Apache Spark | Artificial Intelligence | CI/CDMid-level Full TimeShanghai2d ago