Lead Engineer, Machine Learning
Bjak · Remote · Singapore ·
COMPENSATION
€•• – ••k
Listed on the employer's job board (Ashby)
- Lead ML Engineer
- Fully remote
- Python + PyTorch
- GPU systems scale
- Production focus
Join a small, high-talent-density team building AI-native tools for email, notes, and tasks. The goal is to reduce prompting needs while ensuring high reliability in real-world task completion.
As Lead Engineer, you will bridge research and product by owning the execution layer of intelligence. You will build scalable systems for training, evaluation, and inference.
What you'll do
- Own end-to-end ML systems from data and training to deployment
- Build and evolve pipelines for large model training and fine-tuning
- Design evaluation systems for capability, safety, and performance
- Architect high-performance inference optimizing latency and cost
- Create data pipelines for real-world and synthetic training data
- Establish infrastructure for deploying and monitoring models
- Partner with research and engineering to improve product features
What you'll need
- Experience shipping production ML systems, not just prototypes
- Strong grasp of large-model training, fine-tuning, 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 systems
How you'll work
- Small, hands-on team with broad ownership
- Fast decision-making balancing speed and fundamentals
- Interviews via virtual meetings or onsite
- Prompt hiring decisions after 3-4 interviews