Principal ML Engineer: Production-Grade AI Pipelines

Visa Hunt; Singapore; On-site; SGD 180,000 - 260,000; Full time

Job summary

Our client, a stealth AI startup, is seeking a Principal Machine Learning Engineer to build and deploy production-grade ML systems powering its AI platform. You will translate research into scalable solutions by developing robust training pipelines, inference systems, evaluation frameworks, and deployment infrastructure.

Working closely with research and application engineering teams, you will deliver reliable, high-performance ML systems under production constraints such as latency, cost,

Qualifications

• Strong background in deep learning and transformer-based architectures.
• Hands-on experience training, fine-tuning, or deploying large-scale ML models in production.
• Proficiency with PyTorch or JAX.
• Experience with distributed training and inference frameworks (DeepSpeed, FSDP, Megatron, ZeRO, Ray).
• Strong software engineering skills and production-grade system design.
• Experience optimizing GPU workloads (memory, quantization, mixed precision).
• Ability to own end-to-end ML systems in fast-moving environments.

Responsibilities

• Build and own end-to-end ML pipelines (data processing, training, evaluation, inference, deployment).
• Fine-tune and adapt models using LoRA, QLoRA, SFT, DPO, and distillation.
• Design scalable inference systems balancing latency, cost, and reliability.
• Develop data pipelines for synthetic and real-world datasets.
• Build evaluation frameworks for performance, robustness, safety, and bias.
• Optimize deployments with GPU optimization and scaling strategies.
• Collaborate with backend, mobile, and desktop teams to integrate ML systems.
• Iterate rapidly and monitor real-world performance under production constraints.

Skills

Deep learning

Transformer architectures

Production ML

Software engineering

GPU optimization

Problem solving

Tools

PyTorch

JAX

DeepSpeed

FSDP

Megatron

Ray

Job description

Our client, a stealth AI startup, is seeking a Principal Machine Learning Engineer to build and deploy production-grade ML systems powering its AI platform. You will translate research into scalable solutions by developing robust training pipelines, inference systems, evaluation frameworks, and deployment infrastructure.

Working closely with research and application engineering teams, you will deliver reliable, high-performance ML systems under production constraints such as latency, cost,