Western Digital is looking for a software engineer to build internal tools that power the development of world-class hard disk drives by turning massive amounts of machine-generated data into insights using Machine Learning, GenAI, LLMs, and more.
Requirements
- 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.
- Kafka
- Kubernetes
- Jenkins
- RESTful APIs
- JavaScript
- Node.js
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.
Other
- A recent MS or PhD grad in Computer Science, Computer Engineering, Data Science, or a related field. (Dec 2024 - Dec 2025 Graduates only)
- 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.
- Work on-site in a close-knit, fast-paced, highly collaborative team that values ownership, curiosity, and continuous learning.
