Senior Technologist - Machine Learning
Western Digital16 hours ago
Singapore, SingaporeMid Level / Senior
Responsibilities
- Design and own CNN, U-Net, and ViT systems for precision inspection, measurement, and defect classification.
- Own real-time anomaly detection using sensor and time-series product development data.
- Validate and deploy physics-informed AI, PINNs, surrogate modeling, active learning, and Bayesian experimental design systems.
- Architect and own the ML platform using MLflow, Docker, AWS EKS/Kubernetes, observability tools, model monitoring, and LLM gateways.
- Integrate acquisition functions with feature data, laboratory scheduling systems, and instrument control workflows.
- Lead design reviews and code reviews, provide technical direction, define data interface contracts, and mentor junior team members.
- Partner with domain scientists from problem definition through deployment and ship ML models into product development.
Requirements
- Bachelor's or Master's degree in Artificial Intelligence, Machine Learning, Computer Science, or a related field, or equivalent depth demonstrated through open-source contributions or a significant GitHub portfolio.
- Minimum 3–5 years of hands-on technical experience.
- Demonstrated ownership of AI systems from design through product development deployment.
- Experience directing a small ML or AI engineering team, including design reviews, mentoring, and cross-functional delivery.
- Expert-level Python proficiency and expert PyTorch skills, including custom loss functions and full training-loop ownership.
- Experience with CNN, U-Net, and ViT architectures, real-time anomaly detection, surrogate modeling, active learning, and Bayesian experimental design.
- Experience with product development MLOps, including MLflow, Docker, AWS EKS, observability platforms, and LLM gateways.
- Preferred experience includes materials science, semiconductor or precision product development, PINNs, reinforcement learning, RAG/GraphRAG, RLHF, reward modeling, LLM fine-tuning, LoRA, QLoRA, and LLM distillation.
- Significant open-source ML contributions, public technical writing, or a documented GitHub portfolio demonstrating scientific ML or product development ML engineering depth will be considered.
Tech Stack
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.