Staff ML Engineer: Scalable AI Data Systems

WD; Singapore; On-site; SGD 60,000 - 90,000; Full time

Job summary

WD is building the infrastructure behind the AI-driven data economy. We are looking for engineers to join a focused AI team solving hard scientific problems in precision product development.

You will own two real responsibilities from day one and work under mentorship from engineers and researchers with world-class backgrounds. If you are a fresh graduate or early-career professional, you will learn physics-informed AI, Bayesian methods, and ML systems in your first year while contributing to

Qualifications

• Bachelor's or Master's in AI, ML, CS, Electrical Engineering, Applied Mathematics, or related field.
• Fresh graduate to 1 year; demonstrated ML project competency primary criterion.
• Strong Python, PyTorch, and ML fundamentals required.

Responsibilities

• Experiment Tracking & Evaluation: log MLflow experiments and generate evaluation reports.
• Train CNN-based models for image classification and defect detection under guidance.
• Package models in Docker; assist CI/CD and deployment validation.

Skills

Python

PyTorch

CNN Basics

Surrogate Modeling

Active Learning

MLflow

Docker

Model Evaluation

Learning Mindset

Education

Bachelor's or Master's in AI/ML/CS/EE/Applied Math

Tools

Docker

MLflow

AWS Fundamentals

Job description

WD is building the infrastructure behind the AI-driven data economy. We are looking for engineers to join a focused AI team solving hard scientific problems in precision product development.

You will own two real responsibilities from day one and work under mentorship from engineers and researchers with world-class backgrounds. If you are a fresh graduate or early-career professional, you will learn physics-informed AI, Bayesian methods, and ML systems in your first year while contributing to