Principal Data Scientist, TPG

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

Job summary

Micron is seeking a Lead Engineer for Applied AI and Agentic Systems. You will own architecture and drive delivery of AI systems used by engineers daily. You will turn vague problems into concrete designs, plans, and working software, guiding a team of engineers through deep technical challenges.

The role requires depth in ML plus strong software engineering, cloud deployment, and agentic AI. You’ll set direction, review designs, and ship guardrails for trust in AI systems.

Qualifications

• Bachelor's degree in Computer Science, Data Science, Electrical/Computer Engineering, Statistics, Mathematics, Physics, or related 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.

Responsibilities

• Set the architecture and technical direction for applied AI, ML, and agentic AI work.
• Design systems end to end: data pipelines, features, model development, evaluation, inference services, monitoring.

Skills

Machine learning

Statistical modeling

Python (PyTorch)

Production ML

CI/CD

Education

Bachelor's degree in CS/Data Science/EE/Math/Physics

Tools

PyTorch

TensorFlow

scikit-learn

XGBoost

AWS

GCP

Kubernetes

Docker

Job description

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 Innovation, Technology and Products - Applied AI and Data Science organization builds AI, machine learning, and analytics systems for hard semiconductor engineering problems. We work directly with product engineering, reliability, test, manufacturing, and systems teams. Our engineers pair semiconductor domain knowledge with modern ML engineering, cloud platforms, and agentic AI. Everything we build is meant to run in production, not stop at a prototype.

We are looking for a Lead Engineer, Applied AI and Agentic Systems. You will own the architecture and drive delivery of AI systems that engineers rely on every day. Problems will reach you vague. You will turn them into designs, plans, and working software.

You bring depth in machine learning plus real strength in software engineering, cloud deployment, and agentic AI. You set direction, review designs, question assumptions, and grow the engineers around you. When a problem gets genuinely hard, you dig into the code alongside the team. If that sounds like your kind of work, we would like to meet you!

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.

Minimum 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.

Preferred qualifications

• Master's degree or PhD in Computer Science or a related technical field, or equivalent experience.
• People leadership: mentoring engineers, developing careers, and forming team capability.
• Semiconductors, memory, storage, electronics, hardware systems, or advanced manufacturing.
• AI/ML deployed in cloud, hybrid, or on-prem environments — AWS, GCP, Azure, Kubernetes, OpenShift, or Docker.
• LLMs, RAG, and agentic workflows: agent frameworks, model gateways, knowledge graphs, enterprise search, MCP tool integration, or workflow orchestration.

The US base salary range that Micron Technology estimates it could pay for this full-time position is:

$145,000.00 - $360,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 [e-mail] 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.