Staff Engineer, Full-Stack Data Scientist

Company Description

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.

We are a key partner to some of the largest and highest growth organizations in the world. From energizing the most competitive 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.

Binge-watch any shows, use social media or shop online lately? You’ll find Western Digital supporting the storage infrastructure behind many of these platforms. And, that flash memory card that captures and preserves your most precious moments? That’s us, too.

We offer an expansive portfolio of technologies, storage devices and platforms for business and consumers alike. Our data-centric solutions are comprised of the Western Digital®, G-Technology™, SanDisk® and WD® brands.

Today’s exceptional challenges require your unique skills. It’s You & Western Digital. Together, we’re the next BIG thing in data.

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 policy is available here.

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

Western Digital 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 jobs.accommodations@wdc.com 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.

Based on our experience, we anticipate that the application deadline will be 02/04/2025 (3 months from posting), although we reserve the right to close the application process sooner if we hire an applicant for this position before the application deadline. If we are not able to hire someone from this role before the application deadline, we will update this posting with a new anticipated application deadline.

#LI-KH1

Compensation & Benefits Details

• An employee’s pay position within the salary range may be based on several factors including but not limited to (1) relevant education; qualifications; certifications; and experience; (2) skills, ability, knowledge of the job; (3) performance, contribution and results; (4) geographic location; (5) shift; (6) internal and external equity; and (7) business and organizational needs.
• The salary range is what we believe to be the range of possible compensation for this role at the time of this posting.  We may ultimately pay more or less than the posted range and this range is only applicable for jobs to be performed in California, Colorado, New York or remote jobs that can be performed in California, Colorado and New York.  This range may be modified in the future.
• You will be eligible to participate in Western Digital’s Short-Term Incentive (STI) Plan, 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. Please note that not all roles are eligible to participate in the LTI program, and not all roles are eligible for equity under the LTI plan. RSU awards are also available to eligible new hires, subject to Western Digital’s Standard Terms and Conditions for Restricted Stock Unit Awards.
• We offer a comprehensive package of benefits including 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, and the Western Digital Savings 401(k) Plan.
• Note: No amount of pay is considered to be wages or compensation until such amount is earned, vested, and determinable. The amount and availability of any bonus, commission, benefits, or any other form of compensation and benefits that are allocable to a particular employee remains in the Company's sole discretion unless and until paid and may be modified at the Company’s sole discretion, consistent with the law.