Staff Machine Learning Engineer

BJAK

Full-time Electrical & Energy Engineering
Apply Now
Location
singapore, singapore, Singapore
Posted
June 07, 2026

Job Description

About The Role

A1 is building a proactive AI chat app for everyday users to bring intelligence to conversations, errands, organising and workflows. Unlike traditional chat-based applications, our product focuses on achieving high reliability for long-running workflows, persistent context, and real-world task completion. The system must handle multi-step reasoning, interact with external tools, and remain reliable despite non-deterministic model behavior.

What You'll Do

  • 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.
  • Design and maintain data systems for high-quality synthetic and real-world training data.
  • Implement evaluation pipelines co...