ML Platform Engineer: Build Scalable AI Infrastructure

AI Chopping Block; Singapore; On-site; SGD 180,000 - 260,000; Full time

Job summary

A1 is building proactive AI capabilities and seeks an ML Platform Engineer to power the AI stack. You will design and operate systems behind the AI, from training to deployment, inference, observability, and continuous improvement.

You will collaborate with AI researchers and product engineers to ship reliable, scalable production systems, develop pipelines for data prep, training, evaluation, and model release, and push for faster experimentation and cost efficiency.

Qualifications

• Strong software engineering fundamentals and experience building production systems.
• Experience building ML infrastructure, platforms, or production machine learning systems.
• Experience with model deployment, inference, evaluation, or data pipelines.
• Strong understanding of distributed systems and system reliability.
• Ability to write clean, maintainable, production-quality code.
• Comfortable working in ambiguous, fast-moving environments.
• Bias toward ownership, experimentation, and continuous improvement.

Responsibilities

• Build and operate the ML infrastructure and platforms powering A1’s AI products.
• Design systems for model training, evaluation, deployment, inference, and experimentation.
• Build and optimise model serving and inference infrastructure for high-throughput and low-latency workloads.
• Improve reliability, scalability, latency, and cost efficiency of AI systems.
• Develop reliable pipelines for data preparation, training, evaluation, model release, and continuous improvement.
• Build platforms and tooling that enable AI engineers and researchers to experiment, evaluate, and ship models faster.
• Develop evaluation and benchmarking infrastructure to measure model quality, performance, and regressions.
• Build production observability, monitoring, tracing, and alerting for AI/ML workloads.
• Improve AI systems across reliability, scalability, latency, throughput, and cost.

Skills

Strong software engineering

Production systems

Distributed systems

Ambiguity tolerance

Ownership mindset

Tools

PyTorch/JAX

vLLM/SGLang/TensorRT-LLM

Cloud infrastructure

GPU performance tooling

Vector databases

Job description

A1 is building proactive AI capabilities and seeks an ML Platform Engineer to power the AI stack. You will design and operate systems behind the AI, from training to deployment, inference, observability, and continuous improvement.

You will collaborate with AI researchers and product engineers to ship reliable, scalable production systems, develop pipelines for data prep, training, evaluation, and model release, and push for faster experimentation and cost efficiency.