Western Digital is looking for a Software Development Engineering Engineer to define and build the AI-driven architecture for our Automated HDD Assembly systems. You will design, develop, and maintain AI/ML and GenAI solutions that directly impact manufacturing efficiency.
What You'll Do
- Participate in defining AI-driven Software Architecture and System Architecture for Automated HDD Assembly systems.
- Design, develop, and maintain AI/ML and GenAI solutions for manufacturing use cases like machine diagnostics, process optimization, yield improvement, and anomaly detection.
- Develop and integrate GenAI applications, such as LLM-based agents and knowledge retrieval systems, to support engineering troubleshooting and operational decision-making.
- Build and maintain Data Engineering pipelines for data collection, cleansing, and storage from assembly machines, robotics, sensors, and manufacturing systems.
- Collaborate with Software, Process, and Equipment teams to enable data-driven improvements in yield, downtime, and process stability.
- Manage the full AI model lifecycle from data preparation and training to deployment, monitoring, and continuous improvement.
- Integrate AI solutions with existing Machine Control Software, Automation Frameworks, and Manufacturing Systems.
- Execute experiments, analyze large-scale manufacturing data, and translate insights into actionable engineering recommendations.
- Generate and maintain technical documentation for AI models, data pipelines, software releases, and system specifications.
- Validate and test AI-enabled software releases to ensure robustness, reliability, and manufacturing readiness.
- Serve as a technical AI resource and mentor for junior engineers.
- Maintain effective communication with cross-functional technical teams and stakeholders.
- Comply with 6S activities for disciplined engineering practices and EH&S policies for a safe working environment.
What We're Looking For
- Currently pursuing or recently completed a Bachelor’s degree in Computer Engineering, Computer Science, Data Science, AI Engineering, Mechatronics Engineering, or equivalent experience.
- Experience or strong academic/project background in AI / Machine Learning / GenAI, software development for automation, or data engineering fundamentals.
- Basic knowledge of AI concepts including Machine Learning fundamentals, data preprocessing, model evaluation, and an introduction to Generative AI / LLMs.
- Integration experience with Automation, Robotics, Vision, or Manufacturing systems is relevant.
Nice to Have
- Experience or academic projects in AI / Machine Learning / Deep Learning / GenAI, such as predictive models or LLM-based applications.
- Knowledge of Object-Oriented Programming (OOP) and UML Design.
- Experience with Data Engineering tools or concepts like SQL / NoSQL databases, data pipelines, ETL, or data visualization tools.
- Hands-on experience with Automation or Prototype Systems, including Windows, Linux, RTOS, Arduino or embedded platforms, and PLC systems like Beckhoff, Keyence, or Mitsubishi.
- Exposure to AI-enabled vision systems, image processing, or sensor data analytics.
Technical Stack
- AI / Machine Learning / GenAI
- Object-Oriented Programming (OOP)
- UML Design
- SQL / NoSQL databases
- Data pipelines / ETL
- Data visualization / analytics tools
- Windows, Linux, RTOS
- Arduino / embedded platforms
- PLC systems (Beckhoff, Keyence, Mitsubishi)
- Vision, robotics, and HMI systems programming
Team & Environment
This role is part of our Early Career program, which includes structured onboarding, mentorship, and a development curriculum. You will collaborate closely with Software, Process, and Equipment teams.
WD thrives on the power and potential of diversity. We are committed to an inclusive environment where every individual can thrive through a sense of belonging, respect and contribution.



