Technical Lead (Machine Learning)

Stack

Responsibilities

• ▹Lead the end-to-end execution of machine learning systems, including data pipelines, training workflows, evaluation frameworks, inference architecture, and production deployment
• ▹Fine-tune and optimize models using modern techniques such as LoRA, QLoRA, Supervised Fine-Tuning (SFT), Direct Preference Optimization (DPO), and model distillation
• ▹Design, build, and operate scalable inference systems with a focus on latency, cost efficiency, and reliability
• ▹Develop and maintain data pipelines for both synthetic and real-world training datasets
• ▹Build evaluation frameworks to measure model performance, robustness, safety, and bias in collaboration with research teams
• ▹Optimize production deployments through GPU utilization, memory efficiency, inference optimization, and scaling strategies
• ▹Partner closely with application engineering teams to integrate machine learning systems into backend, desktop, and mobile products
• ▹Continuously improve production systems through rapid iteration, monitoring, and data-driven optimization

Requirements

• ▹Proven experience building and deploying production-grade machine learning systems used by real users
• ▹Strong expertise working with large language models and understanding model behavior, limitations, and failure modes
• ▹Experience developing scalable ML infrastructure, training pipelines, and inference systems
• ▹Strong software engineering skills with the ability to write maintainable, production-quality code
• ▹Experience balancing real-world production constraints, including latency, reliability, scalability, cost, and safety
• ▹Strong ownership mindset with the ability to independently drive technical initiatives from design through deployment
• ▹Excellent communication and collaboration skills, with experience in cross-functional, high-performing engineering teams

Nice to have

• ▹Python
• ▹PyTorch and/or JAX
• ▹GPU-based model training and inference systems

Soft skills

About the company

Our client is a stealth AI startup backed by one of Southeast Asia's leading technology companies, currently building its global founding team. It is developing an AI-native communication platform that simplifies everyday tasks by integrating AI directly into conversations, so users no longer need to switch between multiple applications.