Principal AI​/ML Engineer, Validation Strategy & Test Architecture

Micron Technology is a world leader in innovating memory and storage solutions that accelerate the transformation of information into intelligence, inspiring the world to learn, communicate and advance faster than ever.

The Validation Strategy & Test Architecture (ViSTA) AI/ML team leads the integration of artificial intelligence and machine learning into firmware and hardware validation workflows. The team bridges advanced AI capabilities with practical validation engineering to enable data‑driven decision‑making, accelerate validation cycles, and maintain high product quality at scale.

We are seeking a Principal AI/ML Engineer to join the ViSTA AI/ML team. In this role, you will design, develop, and deploy AI/ML solutions that transform firmware and hardware validation. You will work at the intersection of data science, software engineering, and test engineering to build intelligent systems that predict failures, automate analysis, optimize test coverage, and improve validation efficiency across Micron platforms.

• Develop ML models for automated root cause analysis of validation and test failures
• Build predictive analytics for failure forecasting and risk identification
• Design pattern‑recognition algorithms to identify failure signatures in historical data
• Create validation intelligence dashboards that support data‑driven engineering decisions

• Implement AI‑driven test selection and prioritization based on code change risk
• Develop systems to identify test coverage gaps and redundancies
• Build optimization engines that reduce test runtime while preserving confidence

• Develop AI‑powered test generation from requirements and code changes
• Build adaptive and self‑healing test execution systems using historical performance data
• Integrate Large Language Models (LLMs) into validation workflows to improve efficiency

• Partner with firmware, hardware, and system architecture teams to define validation needs
• Translate validation domain expertise into ML features with validation engineers
• Collaborate with platform teams to deploy and scale AI/ML solutions
• Drive AI adoption across validation teams through tooling, guidance, and enablement

• 8+ years of industry experience with a Bachelor’s degree in Computer Science, Computer Engineering, Data Science, Electrical Engineering, or related technical field
• 7+ years of software development experience, primarily using Python
• Strong foundation in machine learning fundamentals, including model evaluation and feature engineering
• Hands‑on experience with ML frameworks such as scikit‑learn, Tensor Flow, or Py Torch
• Experience deploying and maintaining ML models in production environments

• Master's degree with focus on Machine Learning, Artificial Intelligence, or Data Science
• Experience working with Large Language Models (LLMs) and prompt engineering
• Familiarity with MLOps pipelines, model monitoring, and CI/CD practices
• Experience applying NLP techniques to logs, test data, or technical documentation
• Background in semiconductor, storage, firmware, or hardware validation domains
• Experience building predictive failure or reliability analytics systems

$ - $ a year

Additional compensation may include benefits, bonuses, and equity.

Micron benefits include medical, dental and vision plans, income protection, paid family leave, paid time‑off, and paid holidays.

Micron is proud to be an equal opportunity workplace and is an affirmative action employer. All qualified applicants will receive consideration for employment without regard to race, color, religion, sex, sexual orientation, age, national origin, citizenship status, disability, protected veteran status, gender identity or any other factor protected by applicable federal, state, or local laws.