San Jose, California, United States

Western Digital is hiring an Intern - Data & AI Engineering

Responsibilities

  • Develop and deploy ML-driven pipelines that integrate: Image-based defect detection/classification outputs, Electrical test and parametric data, 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, 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

Requirements

  • Master’s degree in Computer Science or a closely related field
  • 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
  • Experience working with multiple data sources or multi-modal datasets

Nice to Have

  • Exposure to anomaly detection, unsupervised learning, or classification models, including defect detection or quality analysis
  • 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

Additional Information

  • Western Digital is committed to providing equal opportunities to all applicants and employees and will not discriminate against any applicant or employee based on their race, color, ancestry, religion (including religious dress and grooming standards), sex (including pregnancy, childbirth or related medical conditions, breastfeeding or related medical conditions), gender (including a person’s gender identity, gender expression, and gender-related appearance and behavior, whether or not stereotypically associated with the person’s assigned sex at birth), age, national origin, sexual orientation, medical condition, marital status (including domestic partnership status), physical disability, mental disability, medical condition, genetic information, protected medical and family care leave, Civil Air Patrol status, military and veteran status, or other legally protected characteristics. We also prohibit harassment of any individual on any of the characteristics listed above. Our non-discrimination policy applies to all aspects of employment. We comply with the laws and regulations set forth in the "Know Your Rights: Workplace Discrimination is Illegal" poster. Our pay transparency policy is available here.
  • Western Digital thrives on the power and potential of diversity. As a global company, we believe the most effective way to embrace the diversity of our customers and communities is to mirror it from within. We believe the fusion of various perspectives results in the best outcomes for our employees, our company, our customers, and the world around us. We are committed to an inclusive environment where every individual can thrive through a sense of belonging, respect and contribution.
  • Western Digital is committed to offering opportunities to applicants with disabilities and ensuring all candidates can successfully navigate our careers website and our hiring process. Please contact us at jobs.accommodations@wdc.com to advise us of your accommodation request. In your email, please include a description of the specific accommodation you are requesting as well as the job title and requisition number of the position for which you are applying.
  • Based on our experience, we anticipate that the application deadline will be 04/20/2026, although we reserve the right to close the application process sooner if we hire an applicant for this position before the application deadline. If we are not able to hire someone from this role before the application deadline, we will update this posting with a new anticipated application deadline.
Required Skills
Debugging
About company
Western Digital
A global technology company specializing in data storage solutions, providing hard drives, SSDs, and data management technologies for businesses and consumers across various sectors.
All jobs at Western Digital Visit website
Job Details
Department Software Development
Category data
Posted 3 months ago