Technical Lead, Machine Learning
Bjak · Remote · China ·
COMPENSATION
€•• – ••k
Listed on the employer's job board (Ashby)
- Technical Lead, ML
- Fully remote
- Python + PyTorch/JAX
- GPU-based systems
- Production-focused
Join a small, high-talent-density team building AI-native applications for email, notes, and tasks. As Technical Lead, you will bridge research and product by creating reliable, scalable ML systems that minimize prompting and reduce hallucinations.
You will own the end-to-end ML lifecycle, from data and training to inference and deployment. This hands-on leadership role requires strong software engineering fundamentals and experience operating GPU-based workloads at scale.
What you'll do
- Own end-to-end ML systems from data and training to 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 data
- Establish infrastructure for deploying, monitoring, and improving models
- Partner with research and engineering to integrate model capabilities
What you'll need
- Experience shipping production ML systems, 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 results
- 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
- Virtual meetings and/or onsite interviews
Hiring process
- Applications evaluated by technical team members
- Three to four interviews maximum
- Prompt decision process after final interview