Production-Scale ML Systems Architect

WD Media (Singapore) Pte Ltd; Singapore; On-site; SGD 180,000 - 240,000; Full time

Job summary

WD Media (Singapore) Pte Ltd is seeking an ML Technical Lead to make physics-informed AI work in product development, deployed in real systems that drive engineering decisions. You will own CNN/U‑Net/ViT architectures for precision inspection, manage an end-to-end MLOps platform, and collaborate with domain scientists from problem framing to deployment.

This role requires 3–5 years of hands-on AI engineering leadership and a proven track record shipping models into production in a global

Qualifications

• Bachelor's or Master's in AI, ML, CS or related field; AI focus preferred.
• Hands-on ownership of AI systems from design to deployment.
• Experience leading a small ML/AI engineering team.
• Proven track record shipping ML models into production.

Responsibilities

• Design and own CNN, U-Net, and ViT systems for precision inspection, measurement, and defect classification.
• Lead real-time anomaly detection from sensor/time-series data streams.
• Own ML platform: MLflow, Docker, AWS EKS; implement CI/CD and observability.
• Mentor team members and drive technical direction across problem definition to deployment.

Skills

Python

PyTorch

Computer Vision

Anomaly Detection

Surrogate Modeling

Active Learning

MLOps

Education

Bachelor's/Master's in AI/CS

Tools

MLflow

Docker

AWS EKS

LangFuse/PortKey

Job description

WD Media (Singapore) Pte Ltd is seeking an ML Technical Lead to make physics-informed AI work in product development, deployed in real systems that drive engineering decisions. You will own CNN/U‑Net/ViT architectures for precision inspection, manage an end-to-end MLOps platform, and collaborate with domain scientists from problem framing to deployment.

This role requires 3–5 years of hands-on AI engineering leadership and a proven track record shipping models into production in a global