Principal Data Scientist, TPG

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

Benefits offered by this job

Medical, dental, vision

Paid family leave

Robust paid time-off

Paid holidays

Job summary

Micron Technology, Inc. is seeking a Lead Engineer for Applied AI and Agentic Systems to own architecture and deliver AI solutions used by semiconductor engineers daily.

You will shape end-to-end AI systems, drive data pipelines, model development, and production deployment while mentoring engineers and guiding technical direction.

Qualifications

• Led complex AI, ML, data science, or software programs as technical lead.
• Hands-on machine learning and statistical modeling with evaluation.
• Proficient in supervised learning, feature engineering, validation, and deployment.

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

Python

ML engineering

Data science

Software engineering

Leadership

Education

Bachelor's degree in Computer Science or related field

Master's degree or PhD preferred

Tools

PyTorch

TensorFlow

scikit-learn

XGBoost

Cloud platforms

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. For additional information regarding the Benefit programs available, please see the Benefits Guide posted on micron.com/careers/benefits.

• medical, dental and vision plans in all locations
• benefit programs that help protect your income if you are unable to work due to illness or injury
• paid family leave
• robust paid time-off program
• 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.

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.