About this role

Join KLA as an AI/ML Software Engineer in Ann Arbor, MI, to build cutting-edge agentic AI solutions that automate workflows and enhance decision-making across the enterprise. You will collaborate with senior engineers to deliver production-grade AI services, contribute to agent orchestration, and continuously improve system performance and quality. This hands-on role is perfect for builders who enjoy experimenting, learning, and deploying real AI systems.

What we are looking for

6

Build and deploy agent-based AI systems, including autonomous agents
Collaborate on scalable and reliable AI/ML system design and implementation
Support benchmarking and performance tuning for AI models
Ensure AI solutions align with enterprise standards for security and efficiency
Experience with Python, TensorFlow, Scikit-learn, and RAG
Proven track record of deploying AI/ML solutions in production

Skills mentioned

4

PythonCI/CDMachine LearningData Science