Staff Engineer, Full-Stack Data Scientist

Company Description

At Western Digital, we are on a mission to unlock the potential of data so people, companies and organizations everywhere can create what’s next. To fulfill our vision, we are always on the lookout for potential team members who share our passion for solving problems to empower others.

When you join Western Digital, you join a legacy more than 50 years in the making. Across our Western Digital®, SanDisk®, SanDisk® Professional, WD® and WD_BLACK™ brands, we have brought some of the most storied advancements in memory and data storage technology to market—and our best, most innovative work is yet to come.

From energizing gaming platforms, to enabling systems to make cities safer and cars smarter and more connected, to powering the data centers behind many of the world’s biggest companies and public cloud, Western Digital is fueling a brighter, smarter future.

Here’s how you can help.

Job Description

The Design Enablement & Automation team is seeking an experienced full-stack data scientist with strong software engineering, AI and Machine Learning skills to develop data-driven automation tools for improving business and engineering efficiency. You will be involved in all aspects in the solution development process: from data collection, to exploratory data analysis, to deploying machine learning models, data visualizations and generative AI agents to production environments.

ESSENTIAL DUTIES AND RESPONSIBILITIES:

• Implementing, training, fine-tuning and deploying AI/ML models and generative AI assistants
• Setting up automated data pipelines to convert raw logs into structured data sources
• Analyzing and verifying data across multiple big data sources
• Creating interactive web-based data visualizations
• Collaborating with cross-functional teams to understand requirements and deliver data-driven solutions
• Proposing and developing new features and enhancements for existing applications

Qualifications

REQUIRED:

• Bachelor's Degree in Computer Science, Software Engineering, Applied Statistics or equivalent
• 3+ years of professional software development experience in Python, SQL, and JavaScript
• 2+ years of AI/ML development experience
• Strong problem-solving skills and is detail-oriented
• Knowledgeable in data science, machine learning, natural language processing and statistics concepts
• Comfortable working with data across the full data pipeline, including ingestion, preprocessing, exploratory data analysis, feature engineering, model training, evaluation, and deployment
• Exceptional written and verbal communication skills
• Interest in flash, semiconductor, non-volatile memory industry

PREFERRED:

• Masters or PhD candidate in Computer Science, Software Engineering, Applied Statistics or equivalent
• Experience in full-stack web application development
• Experience with using the following technologies:
  • Python: Jupyter, Apache Spark, Pandas, scikit-learn, pytorch, and other data science libraries
  • JavaScript: node.js, d3.js, React
  • Database: MySQL, MongoDB
  • Containerization: Docker, Kubernetes
  • Unix-based environments and shell scripting
  • Cloud environments such as Amazon Web Services
• Hands-on experience with git, Jenkins, JIRA, Confluence, and other software development tools
• Experience in EDA / semiconductor industry is a plus
• Knowledge in physical design, design verification or emulation is a plus

Additional Information

Western Digital is committed to providing equal opportunities to all applicants and employees and will not discriminate against any applicant or employee based on their race, color, ancestry, religion (including religious dress and grooming standards), sex (including pregnancy, childbirth or related medical conditions, breastfeeding or related medical conditions), gender (including a person’s gender identity, gender expression, and gender-related appearance and behavior, whether or not stereotypically associated with the person’s assigned sex at birth), age, national origin, sexual orientation, medical condition, marital status (including domestic partnership status), physical disability, mental disability, medical condition, genetic information, protected medical and family care leave, Civil Air Patrol status, military and veteran status, or other legally protected characteristics. We also prohibit harassment of any individual on any of the characteristics listed above. Our non-discrimination policy applies to all aspects of employment. We comply with the laws and regulations set forth in the "Know Your Rights: Workplace Discrimination is Illegal” poster. Our pay transparency po