Staff Engineer - Machine Learning
Core
Build and validate ML systems for precision product development, focusing on physics-informed AI, Bayesian methods, and defect detection.
Role type
Junior Machine Learning Engineer (IC)
Builds
CNN-based models for image classification and defect detection, ML pipelines, and training data quality validation.
Domain
Manufacturing / AI / Computer Vision
Deliverable
production ML models
Required skills
Python, PyTorch, CNN architecture, Surrogate modeling concepts, Active learning awareness, MLflow, Docker, Model evaluation
Preferred skills
U-Net or ViT, Uncertainty quantification, Bayesian methods, Time-series data, Reinforcement Learning, RAG, AWS
Responsibilities
Log and evaluate ML experiments using MLflow; validate training data quality and flag anomalies; build and train CNN models for defect detection; package models in Docker and support CI/CD scripts.
Seniority
Junior, entry-level (0-1 year experience) (via careerplan.io/jobs/744000153657160-staff-engineer-machine-learning-at-western-digital)