Principal Machine Learning Engineer

Principal Machine Learning Engineer

Our client is a stealth AI startup backed by one of Southeast Asia's leading technology companies and is currently building its global founding team.

The company is developing an AI-native communication platform designed to simplify everyday tasks by integrating AI directly into conversations. Instead of switching between multiple applications, users can plan, organize, compare, research, and complete tasks within a single intelligent assistant.

Serving a market of billions of users still relying on traditional productivity tools, the platform focuses on delivering reliable AI workflows, persistent context, multi-step reasoning, and seamless task execution. The mission is to create an AI assistant that significantly improves productivity while making everyday work simpler and more intuitive.

Our client is seeking a Principal Machine Learning Engineer to build and deploy production-grade machine learning systems that power its AI platform. This role focuses on translating research into scalable solutions by developing robust training pipelines, inference systems, evaluation frameworks, and deployment infrastructure.

Working closely with research and application engineering teams, this position will play a key role in delivering reliable, high-performance ML systems that operate effectively under real-world production constraints.

Key Responsibilities

• Build and own end-to-end machine learning pipelines covering data processing, model training, evaluation, inference, and deployment.
• Fine-tune and adapt models using modern techniques such as LoRA, QLoRA, Supervised Fine-Tuning (SFT), Direct Preference Optimization (DPO), and model distillation.
• Design and operate scalable inference systems while balancing latency, cost, and reliability.
• Develop and maintain data pipelines for both synthetic and real-world training datasets.
• Build evaluation frameworks to assess model performance, robustness, safety, and bias in collaboration with research teams.
• Optimize production deployments through GPU optimization, memory efficiency, latency reduction, and scaling strategies.
• Collaborate with application engineering teams to integrate machine learning systems into backend, mobile, and desktop applications.
• Continuously improve ML systems through rapid iteration and real-world performance monitoring while balancing production constraints such as latency, cost, reliability, and safety.