Principal/Staff ML/AI Engineer, Product Engineering, STPG

Our vision is to transform how the world uses information to enrich life for all.

Join an inclusive team passionate about one thing: using their expertise in the relentless pursuit of innovation for customers and partners. The solutions we build help make everything from virtual reality experiences to breakthroughs in neural networks possible. We do it all while committing to integrity, sustainability, and giving back to our communities. Because doing so can fuel the very innovation we are pursuing.

About the Role:

Join our Product Engineering team to develop AI and Machine Learning solutions that transform semiconductor workflows. This role focuses on model development and agentic AI applications, enabling smarter testing, faster yield advancement, and engineering automation to improve cost, cycle time, and quality. This is a ground-breaking opportunity to work on modern technologies and make a significant impact in a world-class company!

You will also train and guide Citizen Data Scientists (CDS or equivalent experience) within Product Engineering, helping them apply advanced ML techniques and analytics tools to gain domain-specific insights.

Key Responsibilities:

Machine Learning Development:

• Build predictive models for yield, test time, and reliability analysis using structured/tabular data
• Use advanced machine learning techniques to analyze complex datasets and generate actionable insights that improve manufacturing and operational performance
• Build and validate ML workflows for large-scale semiconductor datasets

Agentic AI Solutions:

• Develop intelligent agents to automate engineering tasks (e.g., test program validation, yield analysis, report generation)
• Integrate agents with enterprise tools (JIRA, Confluence, SharePoint) and internal systems.

Data Engineering & Integration:

• Work with cross-functional teams to access, clean, and structure high-volume probe and wafer fabrication inline data to generate insightful information for ML/AI applications
• Collaborate with the inferencing architecture team during deployment phases to ensure successful and balanced production integration

Innovation & Collaboration:

• Find areas where AI can help reduce test time (TTR), improve yield, and automate processes
• Contribute to technical papers, patents, and internal guidelines

Citizen Data Scientist Enablement:

• Guide and upskill Citizen Data Scientists (CDS) in applying advanced ML techniques, guidelines, and analytics tools
• Build structured learning paths and hands-on training modules to strengthen CDS capabilities in machine learning and data-driven problem solving
• Serve as a mentor to offer continuous support, evaluate work, and suggest methods for AI initiatives within Product Engineering.

Requirements

Education:

• Bachelor’s or Master’s degree in Computer Science, Electrical/Electronic Engineering, Data Science, or a related field with at least 5 years of experience in machine learning/AI development and implementation
• PhD in a related field, accompanied by at least 3 years of practical experience in application (or a combination of research and industry involvement)

Core Technical Skills:

• Strong proficiency in Python and data processing libraries (e.g., Pandas, NumPy, Scikit-learn)
• Experience with machine learning algorithms for structured data, including regression techniques, tree-based models, and gradient boosting frameworks
• Ability to build and evaluate ML models for predictive analytics

Additional Skills:

• Experience with machine learning frameworks (e.g., TensorFlow, PyTorch) for processing unstructured data (images, text, logs) and scalable model creation and deployment
• Exposure to agentic AI concepts and frameworks (e.g., LangChain, Microsoft Copilot Studio) for building autonomous workflows and orchestration
• Understanding of LLM-based solutions and their integration into enterprise applications
• Knowledge of MLOps practices, including model lifecycle management, CI/CD for ML, and production inference workflows for robust deployment
• Awareness of distributed computing and cloud-based AI services (e.g., Azure ML, AWS SageMaker) for scalable inferencing.

Soft Skills:

• Strong problem-solving and analytical thinking
• Excellent communication and ability to work in cross-functional teams with semiconductor subject-matter experts for technical exchange and management updates

Preferred Experience:

• Semiconductor industry or manufacturing analytics background
• Hands-on experience with AI agents or workflow automation
• Knowledge of enterprise integration tools (Power Automate, SharePoint connectors)

Why Join Us?

• Drive AI innovation in semiconductor product engineering
• Work on high-impact projects that improve yield, reduce test time, and accelerate time-to-market
• Collaborative environment with opportunities for career growth, patent contributions, and technical leadership

About Micron Technology, Inc.

We are an industry leader in innovative memory and storage solutions transforming how the world uses information to enrich life for all. With a relentless focus on our customers, technology leadership, and manufacturing and operational excellence, Micron delivers a rich portfolio of high-performance DRAM, NAND, and NOR memory and storage products through our Micron® and Crucial® brands. Every day, the innovations that our people create fuel the data economy, enabling advances in artificial intelligence and 5G applications that unleash opportunities — from the data center to the intelligent edge and across the client and mobile user experience.

To learn more, please visit micron.com/careers

All qualified applicants will receive consideration for employment without regard to race, color, religion, sex, sexual orientation, gender identity, national origin, veteran or disability status.

To request assistance with the application process and/or for reasonable accommodations, please contact hrsupport_sg@micron.com

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.