Intern - AI/ML Development Engineer

About the Company

At WD, our vision is to power global innovation and push the boundaries of technology to make what you thought was once impossible, possible. At our core, WD is a company of problem solvers. People achieve extraordinary things given the right technology. For decades, we've been doing just that—our technology helped people put a man on the moon and capture the first-ever picture of a black hole.

We offer an expansive portfolio of technologies, HDDs, and platforms for business, creative professionals, and consumers alike under our Western Digital®, WD®, and WD_BLACK™. We are a key partner to some of the largest and highest-growth organizations in the world. From enabling systems to make cities safer and more connected, to powering the data centers behind many of the world's biggest companies and hyperscale cloud providers, to meeting the massive and ever-growing data storage needs of the AI era, WD is fueling a brighter, smarter future.

Responsibilities

• Assist in developing and testing AI/Machine Learning models for real applications, such as predictive maintenance, data analysis, or automation
• Collect and preprocess data from various sources to support model training and evaluation
• Support the implementation of machine learning algorithms using tools such as Python, PyTorch, or similar frameworks
• Collaborate with engineers and data teams to understand project requirements and contribute to AI-based solutions
• Document development processes, model performance, and findings, and present results to the team or supervisors

Qualifications

• Currently pursuing a Bachelor's or Master's degree in Computer Science, Data Science, Artificial Intelligence, or a related field
• Basic knowledge of programming languages such as Python
• Fundamental understanding of Machine Learning concepts and data analysis
• Familiarity with ML libraries/frameworks (e.g. PyTorch)
• Demonstrate a good level of English proficiency (written/verbal)

Additional Information

WD 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.

WD 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 [email protected] to advise us of your accommodation request.