Tensorflow Pytorch Jobs in Greeley, CO (NOW HIRING)

Our vision is to transform how the world uses information to enrich life for all.
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.
Responsibilities
Validation Intelligence & Analytics

• 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

Test Optimization

• 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

Automation & Infrastructure

• 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

Cross-Functional Collaboration

• 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

Minimum Qualifications

• 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, TensorFlow, or PyTorch
• Experience deploying and maintaining ML models in production environments

Preferred Qualifications

• 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

The US base salary range that Micron Technology estimates it could pay for this full-time position is:
$134,000.00 - $247,000.00 a year
Additional compensation may include benefits, bonuses and equity.
Our salary ranges are determined by role, level, and location. The range displayed on each job posting reflects the minimum and maximum target base pay for new hire salaries of the position across all US locations. Within the range, individual pay is determined by work location and additional job-related factors, including knowledge, skills, experience, tenure and relevant education or training. The pay scale is subject to change depending on business needs. Your recruiter can share more about the specific salary range for your preferred location during the hiring process.
Please note that the compensation details listed in US role postings reflect the base salary only, and do not include bonus, equity, or benefits.
As a world leader in the semiconductor industry, Micron is dedicated to your personal wellbeing and professional growth. Micron benefits are designed to help you stay well, provide peace of mind and help you prepare for the future. We offer a choice of medical, dental and vision plans in all locations enabling team members to select the plans that best meet their family healthcare needs and budget. Micron also provides benefit programs that help protect your income if you are unable to work due to illness or injury, and paid family leave. Additionally, Micron benefits include a robust paid time-off program and paid holidays. For additional information regarding the Benefit programs available, please see the Benefits Guide posted on micron.com/careers/benefits.
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.
To learn about your right to work click here.
To learn more about Micron, please visit micron.com/careers
US Sites Only: To request assistance with the application process and/or for reasonable accommodations, please contact Micron's People Organization at hrsupport_na@micron.com or 1-800-336-8918 (select option #3)
Micron Prohibits the use of child labor and complies with all applicable laws, rules, regulations, and other international and industry labor standards.
Micron does not charge candidates any recruitment fees or unlawfully collect any other payment from candidates as consideration for their employment with Micron.
AI alert: Candidates are encouraged to use AI tools to enhance their resume and/or application materials. However, all information provided must be accurate and reflect the candidate's true skills and experiences. Misuse of AI to fabricate or misrepresent qualifications will result in immediate disqualification.
Fraud alert: Micron advises job seekers to be cautious of unsolicited job offers and to verify the authenticity of any communication claiming to be from Micron by checking the official Micron careers website in the About Micron Technology, Inc.