Principal Engineer - Machine Learning
Western Digital18 hours ago
Singapore, SingaporeStaff+
Responsibilities
- Build, train, evaluate, maintain, and monitor CNN, U-Net, and ViT models for automated inspection and measurement.
- Own real-time anomaly-detection systems for product-development sensor and time-series data, including baselines, threshold calibration, and drift alerting.
- Implement and iterate surrogate-model pipelines and active-learning systems, including acquisition-function configuration and versioned feature-set integration.
- Define the data contract and feature specifications between data engineering and the ML stack, validate datasets, and escalate data-quality issues.
- Maintain model versions, training pipelines, containers, MLflow tracking, monitoring, and CI/CD contributions.
- Provide code-review guidance and document model designs and evaluation outcomes for production handoff.
Requirements
- Bachelor’s or master’s degree in AI, machine learning, computer science, or a related field; an AI major or strong AI research focus is preferred.
- 1–3 years of hands-on ML engineering experience, or equivalent depth demonstrated through internships, academic research, or open-source contributions.
- Demonstrated component-level technical ownership within an end-to-end ML pipeline from training through deployment.
- Strong Python proficiency and proficient-to-expert PyTorch skills for independent model training and evaluation.
- Strong computer-vision foundation in CNNs and hands-on depth in U-Net/segmentation, ViT/transformer-based vision, or time-series anomaly detection.
- Experience with surrogate modeling, active learning, data-to-model interfaces, model evaluation, uncertainty analysis, and technical documentation.
- Practical MLOps experience with MLflow, Docker, Git, and basic CI/CD contributions.
- PINNs, Bayesian neural networks, Gaussian processes, ensemble uncertainty, calibration, reinforcement-learning concepts, AWS machine-learning deployment, or RAG fundamentals are advantageous.
About Western Digital
Western Digital designs and manufactures data storage devices and platforms, including hard disk drives and solid-state drives, for consumer electronics, enterprises, and hyperscale data centers. The public company, founded in 1970 and headquartered in San Jose, sells under the Western Digital and SanDisk brands through OEM and retail channels, and supplies cloud platforms and PC markets worldwide.