Lead Engineer, Machine Learning
Bjak · Remote · Indonesia ·
COMPENSATION
€•• – ••k
Listed on the employer's job board (Ashby)
- Lead ML Engineer
- Fully remote potential
- Python + PyTorch
- GPU-based systems
- Production-focused
You will lead the development of reliable production systems that turn research and model capabilities into real-world products. The focus is on minimizing prompting while organizing conversations, errands, and workflows for billions of users.
The role requires managing the full model lifecycle, including data, training, evaluation, inference, and deployment. You will partner with research and engineering teams to ensure high reliability and reduce hallucinations in long-running tasks.
What you'll do
- Own end-to-end ML systems from data collection to deployment
- Build and evolve training and fine-tuning pipelines for large models
- Design evaluation systems for capability, safety, and performance
- Architect high-performance inference systems optimizing latency and cost
- Create data pipelines for real-world and synthetic training data
- Establish infrastructure for monitoring and improving deployed models
- Partner with research and engineering to integrate model capabilities
What you'll need
- Experience shipping production ML systems, not just prototypes
- Strong understanding of large-model training and inference
- Solid software engineering and systems fundamentals
- Experience operating GPU-based ML workloads at scale
- Ability to navigate ambiguous problems with technical judgment
- Bias toward experimentation, measurement, and shipping code
- High standards for correctness and production quality
Stack
- Python
- PyTorch
- JAX
- GPU-based training and inference systems
How you'll work
- Small, high-talent-density team environment
- Broad ownership and independent execution expected
- Fast decision-making balancing speed and fundamentals
- Interviews conducted via virtual meetings or onsite
Hiring process
- Technical team evaluates initial applications
- Schedule three to four interviews maximum
- Virtual or onsite interview formats available
- Prompt decision process after final interviews