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
-
Ai算法工程师 CNY 144K-240KBig Data | Big data processing | Computer Vision | Data Analysis | Data ProcessingEntry-level Full Time深圳1d ago
-
【校招实习】Ai算法工程师 CNY 25K-37KComputer Vision | Data Analysis | Deep learning | Distributed Computing | Feature EngineeringInternship | Research and development experienceEntry-level Internship深圳1d ago
-
Entry-level Internship深圳1d ago
-
Ai算法暑期实习生(Llm/强化学习) CNY 37K-37KDPO | Deep learning | Fine Tuning | LLM Fine-tuning | Language ModelsAt least 2 months internship duration | Beijing internship location | Full-time internshipEntry-level Internship北京2d ago
-
Ai算法暑期实习生(Llm/强化学习) CNY 37K-37KDPO | Deep learning | Language Models | Large Language Models | Machine LearningFull-time internship | Internship 2 months or moreEntry-level Internship北京2d ago
-
视觉感知算法工程师 - 榫卯计划 CNY 96K-120K4D reconstruction | C++ | Camera | Computer Vision | Computer Vision DatasetsEntry-level Full Time深圳2d ago
-
数据开发工程师 CNY 144K-240KAPI Development | Airbyte | Apache Flink | Apache Kafka | Apache SparkAI tool subscription | API quota | Cloud credits | Fast execution | Flat organizationSenior-level Full Time深圳2d ago
-
Ai算法暑期实习生(Llm/强化学习) CNY 37K-37KDPO | Deep learning | Experimentation | Information Retrieval | Language ModelsContinuous internship 2 months or more | Full-time internship | Location BeijingEntry-level Internship北京2d ago
-
Ai 应用工程师实习生 CNY 36K-37KLangchain | Language Models | Large Language Models | Prompt engineering | PyTorchEntry-level Internship北京2d ago
-
Applied AI Engineer (Agents)-算法2组 CNY 144K-240KApp Development | Context engineering | Data Engineering | Evaluation | Fine TuningMid-level Full Time深圳、厦门2d ago
-
Mid-level Full TimeChina - Hong Kong3d ago
-
CUDA | DeepSpeed | Diffusion Models | Distributed Training | FSDPSenior-level Full TimeBeijing, Beijing, China3d 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, China3d ago
-
Senior-level Full TimeChina, Shanghai3d ago
-
Deep Learning Compiler Engineer - CUDA CNY 144K-240KC# | C++ | CUDA | Computer Architecture | Distributed communicationComprehensive benefitsSenior-level Full TimeChina, Shanghai3d ago
-
Senior AI Engineer CNY 144K-240KCheckpointing | Distributed Training | Efficient Fine Tuning | Fine Tuning | GPU workload optimizationSenior-level Full TimeBeijing Yizhuang, China3d ago
-
Senior-level Full TimeCHN - Guangzhou - EMS, China3d ago
-
IT Developer Engineer (Phython+AI) CNY 300K-420KArtificial Intelligence | Code review | Debugging | Machine Learning | Production SupportFlexible work hours | In-person collaborationMid-level Full TimeCN004 - Shanghai, China (CN004)3d ago
-
Entry-level Full TimeShanghai, China3d ago
-
Entry-level Internship广州3d ago
-
Mid-level Full Time深圳3d ago
-
Mid-level Full Time深圳3d ago
-
优才-具身大模型训练框架工程师-觅蜂子公司 CNY 500K-500KC++ | CPU Optimization | CUDA | CUDA graph | Communication overlapMid-level Full Time上海 R3d ago
-
Robotaxi VLA 大模型算法实习生 CNY 25K-37KAutonomous Driving | C++ | Case Development | Data cleaning | Data labelingEntry-level Internship广州3d ago
-
Applied AI Engineer CNY 300K-399KAutomation | Customer Relationship Management | Customer relationship | Evaluation | FintechEnglish working environment | Fast interview process | Remote workMid-level Full TimeChina3d ago