AI Engineer, Service Supply Chain & Graph ML

KLA; Ann Arbor Charter Township (MI); On-site; USD 90,000 - 133,000; Full time

Job summary

KLA is seeking a skilled AI/ML engineer in Michigan to build predictive tools for spare parts planning and to design graph-based data models for enterprise analytics. You will craft end-to-end solutions from data modeling to deployment, collaborating with engineers and business stakeholders to deliver scalable, explainable insights.

The role emphasizes production-grade tooling, governance, and clear documentation, with a focus on improving forecasting, inventory optimization, and decision

Qualifications

• Strong Python for data/ML development (pandas/numpy)
• Experience building production tools or internal dashboards
• Solid graph theory fundamentals and graph databases
• Experience with data modeling, pipelines, and data quality checks
• Ability to translate business needs into effective tools and enable adoption

Responsibilities

• Build and maintain internal tools, dashboards, and workflows for spares planning analytics
• Define graph data model to represent spares demand, parts, tools, configurations, sites
• Develop predictive models for demand forecasting and supply planning signals
• Establish MLOps patterns: testing, monitoring, deployment, data lineage, docs

Skills

Python

Graph theory

SQL

Tooling usability

Collaboration

Education

MS or PhD in CS/DS/OR

BS + 5+ years in software/ML/data eng

Tools

Neo4j

APIs/Web apps

ML frameworks

Job description

KLA is seeking a skilled AI/ML engineer in Michigan to build predictive tools for spare parts planning and to design graph-based data models for enterprise analytics. You will craft end-to-end solutions from data modeling to deployment, collaborating with engineers and business stakeholders to deliver scalable, explainable insights.

The role emphasizes production-grade tooling, governance, and clear documentation, with a focus on improving forecasting, inventory optimization, and decision