Technical Lead, Machine Learning (AI)

TL;DR

Technical Lead, Machine Learning (AI): Own end-to-end ML system execution for a proactive AI chat app with an accent on reliable, scalable production-grade systems handling multi-step reasoning and real-world tasks. Focus on fine-tuning models with LoRA/QLoRA/SFT/DPO, architecting inference systems, building data pipelines, and integrating into products under latency, cost, and reliability constraints.

Location: Remote (Singapore)

Company

A1 is building a proactive AI chat app for everyday users, focusing on high reliability for long-running workflows, persistent context, and real-world task completion.

What you will do

• Own end-to-end ML system execution: data pipelines, training workflows, evaluation systems, inference architecture, and deployment.
• Fine-tune and adapt models using LoRA, QLoRA, SFT, DPO, and distillation.
• Architect scalable inference systems balancing latency, cost, and reliability; optimize GPU usage and memory efficiency.
• Design data systems for synthetic and real-world training data; implement evaluation pipelines for performance, robustness, safety, and bias.
• Handle production deployment, scaling policies, and issue resolution to minimize user impact.
• Collaborate with application engineering to integrate ML systems into backend, mobile, and desktop products.

Requirements

• Built or shipped real ML systems used by people, not just demos; comfortable with large models and their failure modes.
• Strong production-grade code in Python with PyTorch/JAX; experience with GPU-based training and inference.
• Self-directed, pragmatic ownership of outcomes; clear communication and collaboration in small high-trust teams.

Nice to have

• Experience making pragmatic trade-offs and shipping improvements quickly under production constraints.

Culture & Benefits

• Work in small, high-talent-density, hands-on teams making collective decisions at rapid speed.
• Balance between shipping high-quality work and learning; bring structure, judgment, and independent execution.
• High-trust environment focused on putting magical AI products into users' hands globally.

Hiring process

• Applications evaluated by technical team; 3-4 virtual and/or onsite interviews if fit.
• Prompt decisions with transparency and efficiency; offers for exceptional skills and mindset.