Technical Lead, Machine Learning

Overview

Lead the execution layer of machine learning, turning research and model capabilities into reliable, scalable production systems. Own work across data, training, evaluation, inference, and deployment.

What you'll do

• Own end-to-end ML 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 for latency, GPU utilization, memory, cost, and reliability.
• Build real-world and synthetic training-data pipelines.
• Deploy, monitor, and continuously improve production models.
• Partner with research and application engineering to improve products.

What you'll need

• Experience building and shipping production ML systems.
• Strong understanding of large-model training, fine-tuning, evaluation, and inference.
• Strong software engineering and systems fundamentals.
• Experience operating GPU-based ML workloads at meaningful scale.
• Ability to navigate ambiguous problems independently.
• Proficiency with Python and PyTorch or JAX.

Details

• Location: China.
• Interviews may be virtual or onsite.