Western Digital is hiring a Data & AI Engineering Intern to join our Head Development (HD) organization. In this role, you will focus on building next-generation AI/ML capabilities that accelerate design learning, reduce technical risk, and extract insights from complex, multi-modal engineering data. This is a hands-on position where you will design and implement intelligent systems that connect image-based defect signals with electrical and design data to generate actionable engineering insights.

What You'll Do

Develop and deploy ML-driven pipelines that integrate image-based defect detection/classification outputs, electrical test and parametric data, and design and process metadata.
Build intelligent agents or analytical tools that identify relationships between defects and performance, detect emerging failure modes using multi-modal data, and generate hypotheses to accelerate root cause analysis.
Design data models and workflows to support scalable analytics across experiments and design iterations.
Collaborate with engineers and data scientists to translate domain problems into ML/AI solutions.
Present findings, insights, and prototypes to technical stakeholders.

What We're Looking For

Currently pursuing a Master’s degree in Computer Science or a closely related field.

Nice to Have

Strong Python programming skills, with experience building data-driven tools, models, or applications.
Understanding of machine learning fundamentals, such as classification, clustering, model training, and evaluation.
Experience working with data analysis or data pipelines using tools such as Pandas, NumPy, Dataiku, or similar technologies.
Familiarity with software engineering best practices, including writing maintainable code, debugging, and version control.
Experience or coursework related to computer vision, such as convolutional neural networks (CNNs) or image embeddings.
Exposure to anomaly detection, unsupervised learning, or classification models, including defect detection or quality analysis.
Experience working with multiple data sources or multi-modal datasets.
Experience building AI agents or applications using large language models (LLMs).
Understanding of experimental design, model evaluation, or statistical analysis.
Interest in working with manufacturing, semiconductor, or hardware-related datasets.
Curiosity and strong problem-solving skills, especially in research-oriented environments where experimentation and iteration are part of the process.

Technical Stack

Python, Pandas, NumPy, Dataiku
Machine Learning, Computer Vision
CNNs, LLMs

Team & Environment

This role sits within the Head Development (HD) organization at Western Digital.

Benefits & Compensation

Paid vacation time and sick leave
Medical, dental, and vision insurance
Life, accident and disability insurance
Tax-advantaged flexible spending and health savings accounts
Employee assistance program
Voluntary benefit programs (supplemental life and AD&D, legal plan, pet insurance, critical illness, accident and hospital indemnity)
Tuition reimbursement
Transit benefits
The Applause Program
Employee stock purchase plan
WD Savings 401(k) Plan

Compensation includes a salary range applicable for jobs performed in California, Colorado, New York or remote jobs that can be performed in those states. This role is also eligible to participate in the annual Long-Term Incentive (LTI) program (RSUs or cash equivalents), and eligible new hires may receive RSU awards.

Work Mode

This role can be performed in California, Colorado, or New York.

Western Digital is committed to providing equal opportunities to all applicants and employees and will not discriminate based on legally protected characteristics.