Staff ML Engineer: Early-Career in Physics-Driven AI

WESTERN DIGITAL (SINGAPORE) PTE. LTD.; Singapore; On-site; SGD 48,000 - 60,000; Full time

Job summary

WESTERN DIGITAL (SINGAPORE) PTE. LTD. invites fresh graduates to join a focused AI team tackling hard scientific problems in precision product development. You will own real responsibilities from day one and work under mentors with advanced research backgrounds.

You will gain hands-on experience with MLflow experiment tracking, data quality validation, CNN-based model development, and containerized deployment in Docker, including model evaluation and CI/CD support.

Qualifications

• Bachelor's or Master's degree in AI, ML, CS, EE, Applied Mathematics, or related field; AI major or research focus preferred.
• Fresh graduate to 1 year; demonstrated ML project competency is primary criterion; strong final-year project or thesis with ML component; research internship preferred.
• Must have Python, PyTorch, CNN basics, surrogate modeling concepts, active learning awareness, MLflow basics, Docker basics, model evaluation, and a learning mindset.

Responsibilities

• Experiment Tracking & Evaluation Reporting: own MLflow experiment logging, run evaluations, produce reports informing model iterations.
• Training Data Quality Validation: validate training data with feature distribution checks, label verification, anomaly flagging.
• Deep Learning Model Contribution: build/train CNN-based models for image classification/defect detection and participate in evaluation cycles.
• ML Pipeline Contribution: package models in Docker; contribute to CI/CD scripts; run inference tests and support deployment validation.

Skills

Python

PyTorch

CNN Architecture Basics

Surrogate Modeling Concepts

Active Learning Awareness

MLflow Basics

Docker Basics

Model Evaluation

Learning Mindset

Education

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

Tools

Docker

MLflow

AWS fundamentals

Job description

WESTERN DIGITAL (SINGAPORE) PTE. LTD. invites fresh graduates to join a focused AI team tackling hard scientific problems in precision product development. You will own real responsibilities from day one and work under mentors with advanced research backgrounds.

You will gain hands-on experience with MLflow experiment tracking, data quality validation, CNN-based model development, and containerized deployment in Docker, including model evaluation and CI/CD support.