Staff Engineer - Machine Learning
Western Digital2 days ago
Singapore, SingaporeStaff+
Responsibilities
- Own MLflow experiment logging and conduct model evaluations using standard metrics.
- Validate training datasets through feature distribution checks, label verification, and anomaly flagging.
- Build and train CNN-based models for image classification and defect detection.
- Contribute to model evaluation cycles, configuration comparisons, and training-run analysis.
- Package models in Docker, contribute to CI/CD scripts, run inference tests, and support deployment validation.
Requirements
- Bachelor's or Master's degree in AI, Machine Learning, Computer Science, Electrical Engineering, Applied Mathematics, or a related field; an AI major or strong research focus is preferred.
- Fresh graduate to one year of experience; work experience is not required, but demonstrated ML project competency, a strong final-year project or thesis, and preferably a research internship are expected.
- Strong Python proficiency, including clean ML code and NumPy/Pandas basics.
- Foundational ability to build, train, and evaluate a neural network with PyTorch and implement a basic CNN image classifier.
- Understanding of surrogate modeling, active learning, standard model evaluation metrics, and MLflow experiment logging.
- Ability to write a Dockerfile to containerize a Python/ML application.
- Preferred exposure includes U-Net or ViT, uncertainty quantification, Bayesian methods, time-series or sensor data, reinforcement learning, RAG or LLM projects, and AWS fundamentals.
Benefits
- The role is based in the Singapore Office.
Categories
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.