Staff Machine Learning Engineer

Overview

Own the execution layer of machine learning systems, translating research and model capabilities into reliable, scalable production systems. Work hands-on across data, training, evaluation, inference, and deployment.

What you'll do

• Own end-to-end machine learning systems from data and training through deployment.
• Build training and fine-tuning pipelines for large models.
• Design evaluations for capability, robustness, safety, and product performance.
• Architect inference systems while optimizing latency, GPU utilization, memory, cost, and reliability.
• Build real-world and synthetic training-data pipelines.
• Establish infrastructure for deploying, monitoring, and improving models.
• Partner with research and application engineering to deliver product improvements.

What you'll need

• Experience building and shipping production machine learning systems.
• Strong understanding of large-model training, fine-tuning, evaluation, and inference.
• Strong software engineering and systems fundamentals.
• Experience operating GPU-based machine learning workloads at scale.
• Strong technical judgment and ability to work independently through ambiguity.
• Experience with Python.
• Experience with PyTorch or JAX.

Details

• Location: China.
• Interviews are conducted virtually and/or onsite.