WD is a leading company in the Computer Hardware industry, known for its innovative technology solutions. They are seeking a Software Engineer to build and maintain internal software tools that support the development of hard disk drives, utilizing machine learning and data analysis to drive insights.

Responsibilities

• Build and maintain custom internal software tools used to process, analyze, and visualize HDD development and production data
• Develop ETL systems, web-based dashboards, ML models, and analytics pipelines that help product engineers innovate
• Contribute to GenAI/LLM use cases—helping select and implement base models and embedding strategies for real-world deployment
• Design prompt and token-based GenAI solutions that support classification, pattern recognition, and trend visualization
• Work hands-on with modern frameworks (PyTorch, TensorFlow), tools (Python, C/C++, Node.js, Kafka), and databases (MongoDB, MySQL, Elasticsearch)
• Join in on architecture discussions, code reviews, and collaborative problem solving
• Create internal guides and documentation that make your tools accessible to other engineers
• Work on-site in a close-knit, fast-paced, highly collaborative team that values ownership, curiosity, and continuous learning

Skills

• Currently enrolled or recently graduated with an MS or PhD grad in Computer Science, Computer Engineering, Data Science, or a related field
• Right to work in the US without requiring Western Digital to commence or sponsor an immigration case in order to employ you at any point in the future required
• Experience with Python (Jupyter), C/C++, and at least one ML or AI framework (e.g., PyTorch, TensorFlow)
• Completed coursework or hands-on projects involving GenAI, LLMs, or machine learning
• Comfort working with both relational and non-relational databases
• Bonus if you've dabbled in: Kafka, Kubernetes, Jenkins, RESTful APIs, JavaScript, Node.js

Benefits

• Paid vacation time
• Paid sick leave
• Medical/dental/vision insurance
• Life, accident and disability insurance
• Tax-advantaged flexible spending and health savings accounts
• Employee assistance program
• Other voluntary benefit programs such as supplemental life and AD&D, legal plan, pet insurance, critical illness, accident and hospital indemnity
• Tuition reimbursement
• Transit
• The Applause Program
• Employee stock purchase plan
• The WD Savings 401(k) Plan
• You will be eligible to be considered for bonuses under either WD’s Short Term Incentive Plan (“STI Plan”) or the Sales Incentive Plan (“SIP") which provides incentive awards based on Company and individual performance, depending on your role and your performance.
• You may be eligible to participate in our annual Long-Term Incentive (LTI) program, which consists of restricted stock units (RSUs) or cash equivalents, pursuant to the terms of the LTI plan.
• RSU awards are also available to eligible new hires, subject to WD's Standard Terms and Conditions for Restricted Stock Unit Awards.

Company Overview