Staff Machine Learning Engineer
BjakInsurance Comparison company
Singapore, SingaporeLead
BASS Investment
Data & AI
About the role
TL;DR
Owns the execution layer of AI intelligence, translating research into production-grade ML systems.
- •Bjak is building a proactive smart assistant to bring intelligence to everyday applications, focusing on high reliability for long-running workflows and real-world task completion with minimal prompting.
- •Key Responsibilities Own end-to-end ML system execution: data pipelines, training workflows, evaluation systems, inference architecture, and deployment.
- •Fine-tune and adapt models using state-of-the-art methods such as LoRA, QLoRA, SFT, DPO, and distillation.
- •Architect and operate scalable inference systems, balancing latency, cost, and reliability.
- •Implement evaluation pipelines covering performance, robustness, safety, and bias.
- •Own production deployment, including GPU optimization, memory efficiency, latency reduction, and scaling policies.
- •Requirements Experience building and shipping real ML systems used by people.
- •Comfort working with large models and understanding their failure modes.
- •Strong, production-grade coding skills and care for system correctness.
- •Self-directed, pragmatic, and ownership-driven.
- •Clear communication and collaboration in small, high-trust teams.
Required skills
PythonPyTorchJAX
Domain expertise
productivityai
Tech stack
PythonPyTorchJAX