Staff Machine Learning Engineer
Bjak · Remote · Singapore ·
COMPENSATION
€•• – ••k
Listed on the employer's job board (Ashby)
- Staff level role
- Fully remote
- Python + PyTorch
- GPU systems focus
- High ownership
Join a team building proactive, AI-native applications for email, notes, and tasks that require minimal prompting. As a Staff Machine Learning Engineer, you will own the execution layer of intelligence, bridging the gap between research capabilities and scalable production systems.
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, 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 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 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 independently
- Bias toward experimentation, measurement, and shipping
- High standards for correctness and production quality
Stack
- Python
- PyTorch
- JAX
- GPU-based systems
How you'll work
- Small, high-talent-density team
- Broad ownership and independent execution
- Fast decision-making process
- Balance of speed and engineering fundamentals
Hiring process
- Technical team evaluates applications
- Three to four interviews maximum
- Virtual meetings or onsite options
- Prompt decision after final interview