Intern - AI Engineering (Studying Bachelor Degree)

At Western Digital, 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, Western Digital 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®, WD_BLACK™, and SanDisk® Professional brands.

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, Western Digital is fueling a brighter, smarter future.

Today’s exceptional challenges require your unique skills. Together, we can build the future of data storage.

ESSENTIAL DUTIES AND RESPONSIBILITIES:

• Model Development & Training: Design and implement machine learning and deep learning models (e.g., LLMs, Computer Vision, NLP) to address specific use cases.
• Pipeline Architecture: Build and maintain scalable data ingestion and preprocessing pipelines to ensure high-quality data availability for training and inference.
• Deployment & MLOps: Deploy models into production environments (cloud or edge) and manage the MLOps lifecycle, including version control, continuous integration/continuous deployment (CI/CD), and model monitoring.
• Performance Optimization: Optimize model architecture and code for latency, throughput, and cost-efficiency without sacrificing accuracy.
• Collaboration: Work closely with Data Scientists to refine algorithms and with Software Engineers to integrate AI capabilities into core products.
• Research & Innovation: Stay current with the latest advancements in AI/ML frameworks (e.g., PyTorch, TensorFlow) and emerging techniques (e.g., RAG, fine-tuning) to drive innovation.
• Support department activities.
• Work to meet schedules.

This position is part of our Early Career program at WD. Our Early Career program is designed to support individuals beginning their professional career by providing the foundational training through a structured onboarding, mentorship, and development curriculum.

REQUIRED:

• Education: Currently pursuing Bachelor’s degree in Computer Science, Artificial Intelligence, Data Science or equivalent experience

SKILLS:

• Programming Languages: Expert-level proficiency in Python.
• ML Frameworks: Strong hands-on experience with frameworks such as PyTorch, TensorFlow, Keras, or Scikit-learn.
• Cloud Platforms: Experience deploying solutions on cloud platforms like AWS, Google Cloud Platform.
• MLOps: Proficiency with containerization (Docker, Kubernetes) and orchestration tools.
• Database Skills: Strong knowledge of SQL and experience with vector databases or NoSQL.
• The internship period must be least 3 months, Cooperative program intern is preferred​.

#LI-BI1 

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

Notice To Candidates: Please be aware that WD and its subsidiaries will never request payment as a condition for applying for a position or receiving an offer of employment. Should you encounter any such requests, please report it immediately to WD Ethics Helpline or email [email protected].