Machine Learning Engineer (Training Optimization）
Beijing, Beijing, China
CNY 50K-50K (estimate) Entry-level Full Time
Tasks
- Collaborate with research and modeling teams
- Debug and profile training workflows
- Design large scale machine learning training systems
- Evaluate and apply distributed training best practices
- Fine tune training scalability
- Implement and optimize distributed training systems
- Improve GPU utilization and memory efficiency
- Optimize low level performance with CUDA or Triton kernels
- Reduce communication overhead
Perks/Benefits
- N/A
Skills/Tech-stack
CUDA | DeepSpeed | Distributed Training | FSDP | Gradient Checkpointing | JAX | Low Precision | Megatron-LM | NVIDIA Nemo | PyTorch | Python | Triton | Zero
Education
N/A
Related jobs
-
Bash | Data Ingestion | Data Processing | Docker | GCPAsynchronous work culture | Friendly laid-back atmosphereMid-level Full TimeShanghai, China7h ago
-
Computer Graphics | Computer Vision | CoreML | Deep learning | Diffusion ModelsSenior-level Full TimeBeijing, Beijing, China10h ago
-
Entry-level Internship深圳18h ago
-
Entry-level InternshipBeijing,Beijing,China1d ago
-
(Sr) Cloud & Data Engineer CNY 192K-240KAWS | Automation | CI/CD | Container Security | Data ModelingMid-level Full TimeBeijing, Beijing, CN1d ago
-
[Growth Engineering] Staff Back-end Engineer I CNY 144K-240KAnomaly Detection | CI/CD | Containerization | FastAPI | Graph DatabaseSenior-level Full TimeShanghai, China1d ago
-
[Growth Engineering] Staff Back-end Engineer I CNY 144K-240KCI/CD | Containerization | FastAPI | Graph Database | Inference ServerSenior-level Full TimeShanghai, China1d ago
-
Deep Learning Performance Architect CNY 152K-240KComputer Architecture | Deep learning | Inference | JAX | Language ModelsSenior-level Full TimeChina, Shanghai1d ago
-
C# | C++ | Computer Vision | Debugging | Deep learningSenior-level Full TimeChina, Shanghai1d ago
-
Entry-level Internship Part TimeShanghai - Daning Main Blg, China1d ago
-
Entry-level Full Time广州1d ago
-
Senior-level Full Time上海、北京1d ago
-
Mid-level Full Time北京 R1d ago
-
A/B | A/B Experimentation | Autoscaling | Caching | Canary testingCommute subsidy | Disability insurance | Employee stock ownership | Generous vacation | Health insuranceSenior-level Full TimeShanghai, China2d ago
-
Apache Airflow | Apache Flink | Apache Spark | Automated testing | Data LakeCommute subsidy | Competitive retirement pension plans | Employee resource groups | Employee stock ownership | Generous vacation personal daysSenior-level Full TimeShanghai, China2d ago
-
Airflow | CUDA | Data Lake | Data Warehouse | FlinkCommute subsidy | Competitive retirement pension plans | Employee resource groups | Employee stock ownership | Generous vacation personal daysSenior-level Full TimeShanghai, China2d ago
-
A/B | A/B Testing | Autoscaling | B testing | Canary testingCommute subsidy | Competitive retirement pension plans | Employee resource groups | Employee stock ownership | Generous vacationSenior-level Full TimeShanghai, China2d ago
-
Senior Data Engineer, Content Management Systems (China) CNY 144K-240KAPI Integration | AWS | Access Control | Alibaba Cloud | CI/CDAnnual medical check-up | Flexible benefits | Long service award | Medical and life insurance | Paid time offSenior-level Full TimeChina - Shanghai2d ago
-
AWS | Apache Airflow | Apache Kafka | Apache Spark | AzureMid-level Full TimeCN-Shenzhen-HyQ, China2d ago
-
Entry-level Internship Part TimeShanghai (JingAn), China2d ago
-
Senior Data Engineer, Content Management Systems (China) CNY 144K-240KAPI Integration | AWS | Alibaba Cloud | Apache Kafka | Apache SparkAnnual Medical Checkup | Flexible benefits | Life insurance | Long service award | Medical insuranceSenior-level Full TimeChina - Shanghai2d ago
-
Mid-level Full Time深圳2d ago
-
Entry-level Internship北京2d ago
-
Senior Software Engineer - Machine Learning CNY 360K-600KData Analysis | Data Visualization | Deep learning | Experimentation | Fraud DetectionCareer progression | Collaborative culture | Competitive compensation | Global growth opportunitiesSenior-level Full TimeShenzhen, China3d ago
-
AI intern CNY 28K-50KAutomated testing | Continuous integration | Deep learning | Generative AI | JavaEntry-level InternshipBeijing,Beijing,China3d ago