Principal AI Systems Architect for Production ML

Micron Technology, Inc; San Jose (CA); On-site; USD 145,000 - 360,000; Full time

Job summary

Micron Technology, Inc in the United States is seeking a Lead Engineer, Applied AI and Agentic Systems to own the architecture and drive delivery of AI systems used by semiconductor engineers daily.

You will lead designs, review architectures, and guide software engineers, translating vague problems into concrete roadmaps and production-ready solutions with cloud deployment and agentic workflows. This role emphasizes hands-on ML, coding, and collaboration across product engineering teams.

Qualifications

• Bachelor's degree in Computer Science, Data Science, Electrical/Computer Engineering, Statistics, Mathematics, Physics, or a related technical field.
• 8 years of relevant experience, or equivalent practical experience.
• Led complex AI, ML, data science, or software engineering programs as technical lead, directing the work of other engineers.
• Hands-on machine learning and statistical modeling: supervised learning, feature engineering, validation, performance evaluation, and reading model behavior.
• Python fluency with PyTorch, TensorFlow, scikit-learn, or XGBoost. Shipped models into production — inference pipelines, APIs, CI/CD, monitoring.

Responsibilities

• Set the architecture and technical direction for our applied AI, ML, and agentic AI work.
• Design systems end to end: data pipelines, features, model development, evaluation, inference services, and monitoring.
• Build the models engineers act on — prediction, diagnosis, optimization, and decision support for semiconductor applications.
• Ship agentic AI for engineering analysis, knowledge retrieval, workflow automation, and decision support. Define the guardrails and evaluations that earn trust in it.
• Break wide-open challenges into roadmaps, turning points, and success criteria. Report progress, risk, and model limits plainly to engineers and executives.

Skills

Led AI/ML programs

Machine learning & statistics

Python & ML libraries

Software engineering

Cloud deployment

Agentic AI design

Education

Bachelor's degree in Computer Science / related field

Master's degree or PhD in related field

Tools

PyTorch

TensorFlow

scikit-learn

XGBoost

Docker

Kubernetes

OpenShift

CI/CD

Job description

Micron Technology, Inc in the United States is seeking a Lead Engineer, Applied AI and Agentic Systems to own the architecture and drive delivery of AI systems used by semiconductor engineers daily.

You will lead designs, review architectures, and guide software engineers, translating vague problems into concrete roadmaps and production-ready solutions with cloud deployment and agentic workflows. This role emphasizes hands-on ML, coding, and collaboration across product engineering teams.