Lead Engineer, Machine Learning
Bjak · Remote · China ·
COMPENSATION
€•• – ••k
Listed on the employer's job board (Ashby)
- Lead Machine Learning Engineer
- Fully remote
- Python + PyTorch/JAX
- GPU-based systems
- Production-focused ML
Join a small, high-talent-density team dedicated to creating AI-native applications that reduce the need for complex prompting. As a Lead Engineer, you will bridge the gap between research and product by turning model capabilities into reliable, scalable production systems for billions of users.
What you'll do
- Own end-to-end ML systems from data and training to inference and deployment
- Build and evolve training and fine-tuning pipelines for large models
- Design evaluation systems for capability, robustness, and safety
- Architect high-performance inference systems optimizing latency and cost
- Create data pipelines for high-quality real-world and synthetic training data
- Establish infrastructure for deploying, monitoring, and improving models
- Partner with research and engineering to integrate model capabilities into products
What you'll need
- Experience shipping ML systems in production, not just research prototypes
- Strong understanding of large-model training, fine-tuning, and inference
- Solid software engineering and systems fundamentals
- Experience operating ML workloads at scale on GPU-based systems
- Ability to navigate ambiguous problems with strong technical judgment
- Bias toward experimentation, measurement, and shipping code
- High standards for correctness, reliability, and production quality
Stack
- Python
- PyTorch
- JAX
- GPU-based training and inference systems
How you'll work
- Small, hands-on team with broad ownership
- Fast decision-making balancing speed and engineering fundamentals
- Interviews conducted via virtual meetings or onsite
Hiring process
- Technical team evaluates applications
- Three to four interviews maximum
- Prompt decision process after final interview