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-240KApache Hadoop | Apache Hive | Apache Spark | Deep learning | Feature EngineeringEntry-level Full Time深圳21h ago
-
Senior AI Operations Engineer CNY 144K-240KAlert Correlation | Docker | Forecasting | Function Calling | KubernetesSenior-level Full TimeChina, China2d ago
-
Computer Vision | CoreML | Diffusion Models | GLSL | Geometric processingSenior-level Full TimeBeijing, China2d ago
-
Agentic Engineering | Apache Airflow | Apache NiFi | Automated testing | BigQueryMid-level Full TimeGuangzhou, Guangdong, China2d ago
-
Senior Consultant Specialist(GenAI Agent platform) CNY 160K-240KAWS | Access Management | Agent Orchestration | Audit Logging | AzureSenior-level Full TimeGuangzhou, Guangdong, China R2d ago
-
Apache Airflow | Apache Kafka | Apache NiFi | Automated testing | BigQueryMid-level Full TimeGuangzhou, GD, CN, 5106202d ago
-
Mid-level Full TimeBJN01 - DXC Beijing LSH Center …2d ago
-
Mid-level Full TimeShanghai Jing'An Office, China2d ago
-
3D Reconstruction | Accuracy Metrics | C++ | Camera Calibration | Cloud processingSenior-level Full TimeSuzhou, China2d ago
-
腾讯云数据迁移工程师 CNY 144K-240KAWS | Azure | CI/CD | CloudFormation | Data GovernanceFlexible work arrangements | In-person collaborationMid-level Full TimeSHC01 - DXC Shanghai Campus Phase …2d ago
-
数据平台开发工程师 CNY 180K-300KData Lake | Data Warehouse | Data Warehouse Modeling | Data pipeline | DataOpsMid-level Full Time广州2d ago
-
Senior Software Engineer (GenAI Agent platform) CNY 160K-240KGo | Java | Langchain | Langgraph | PythonSenior-level Full TimeGuangzhou, Guangdong, China R3d ago
-
Biomedical Data | Data Governance | Data Mining | Data Pipelines | Data QualityMid-level Full TimeSuzhou3d ago
-
Mid-level Full TimeShanghai, China3d ago
-
Senior Machine Learning Engineer CNY 400K-510K3D Reconstruction | Adversarial Networks | C++ | Computer Graphics | Computer VisionSenior-level Full TimeShanghai, China3d ago
-
Sr. IT Specialist,AI. CNY 240K-480K.Net Core | AI Search | Agent Framework | Agent systems | AutogenSenior-level Full TimeShanghai, China3d ago
-
Mid-level Full TimeCN-OCG International Center, Cheng Du, China3d ago
-
Senior-level Full TimeShanghai3d ago
-
Senior-level Full Time深圳3d ago
-
Bash | Data Processing | Docker | GCP | Infrastructure as CodeAsynchronous work culture | Friendly work environment | Hands-off management | Opportunity to make impactMid-level Full TimeBeijing, China4d ago
-
Senior Data Scientist CNY 156K-240KExcel | Power BI | Projection | Python | REmployee Assistance Program (EAP) | Flexible working environment | LinkedIn Learning | Volunteer time offSenior-level Full TimeShanghai, SH, China4d ago
-
Machine Learning Engineer CNY 144K-240KData Curation | Data Quality | Deep learning | Fine Tuning | JAXSenior-level Full TimeBeijing4d ago
-
AWS | Apache Airflow | Apache Doris | Apache Iceberg | Apache SparkSenior-level Full TimeShanghai4d ago
-
Artifact correction | C# | C++ | Computed Tomography | Deep learningIn-person work at least 3 days per weekMid-level Full TimeShenyang - PIC, China4d ago
-
Senior Manager-AI Algorithm CNY 144K-240KAPI Development | Agent Orchestration | Agentic AI | CI/CD | Cloud PlatformsSenior-level Full TimeChina4d ago