Interested in this AI Software Engineer role at KLA?
Apply Now →Skills & Technologies
About This Role
To make electronics, you need chips, wafers, transistors, reticles, and... To make these, you must see, test and manufacture them at scale—faster and better than ever before. That's where KLA comes in. Whether you're early in your career or an experienced professional, you'll solve complex challenges, work alongside brilliant minds and help shape the future of technology.
Group/Division
KLA's IT group supports business growth and productivity by connecting people, process and technology around the world. We work to improve the technology that drives our business to thrive and focus on empowering employee use of technology. This integrated approach to customer service, creativity and technological excellence enhances employee productivity, business analytics and process excellence. What You'll Do
In this role, you will play a key part in advancing business priorities by delivering high-impact work across your area of expertise.
We are seeking an AI/ML Engineer to help build agentic AI solutions that automate workflows and support better decision‑making across the enterprise. In this role, you’ll collaborate with senior engineers and product partners to deliver reliable, production‑grade AI services , contribute to agent orchestration approaches, and continuously improve evaluation and monitoring for quality and performance.
The ideal candidate brings proven software engineering fundamentals and experience with modern AI frameworks. This is a hands‑on builder role focused on collaborative design, implementation, and deployment of agentic AI systems. If you enjoy building, experimenting, and learning while deploying real AI systems, we’d love to hear from you! This role will be based at our Midwest HQ in Ann Arbor, MI
Key Responsibilities
- Build and deploy agent‑based AI systems , including autonomous and semi‑autonomous agents that can reason, plan, use tools, and collaborate effectively.
- Support benchmarking efforts for model performance across different deployment environments.
- Collaborate in the design and implementation of scalable and reliable AI/ML systems that support evolving workloads.
- Help ensure solutions align with enterprise standards for security, compliance, observability, and operational efficiency.
- Run focused experiments (e.g., A/B tests, workflow refinements) to improve agent performance.
- Work closely with cross‑functional partners to deliver impactful business outcomes.
Preferred Qualifications
- Experience working with enterprise AI/ML platforms or managed AI services.
- Exposure to model evaluation, monitoring, or performance tuning in production settings.
- Familiarity with MLOps practices and modern deployment workflows.
Minimum Qualifications
- Bachelor's degree in Computer Science, Data Science, AI, or a related field. Master's and/or PhD preferred
- Minimum of 3-5 years of experience in AI/ML, with a proven track record of deploying solutions in a production environment.
- Proficiency in Python, or similar languages, with experience in libraries like TensorFlow, Scikit-learn, etc. Experience with RAG is essential.
- Strong background in statistics, machine learning algorithms, deep learning, neural networks, and system architecture.
- Strong understanding of cloud platforms, DevOps practices, and data processing technologies.
- Demonstrated ability to lead technical projects and work effectively in matrix environments.
- Ability to communicate technical concepts to non-technical team members and business partners.
- Familiarity with software engineering best practices, including version control (Git), CI/CD pipelines, agile methodologies, and system design patterns.
About KLA
We provide advanced inspection tools, metrology systems, process solutions, and computational analytics that make electronics possible, tackling complex challenges. From electron and photon optics to machine learning and data analytics, we seek perfection at the most fundamental level of matter in the universe. If you want to make electronics that push industries forward and make the world a better place, join us.
Total Rewards
Base Pay Range: $105,900.00 - $180,000.00 Annually
Primary Location: USA-MI-Ann Arbor-KLA KLA’s total rewards package for employees may also include participation in performance incentive programs and eligibility for additional benefits including but not limited to: medical, dental, vision, life, and other voluntary benefits, 401(K) including company matching, employee stock purchase program (ESPP), student debt assistance, tuition reimbursement program, development and career growth opportunities and programs, financial planning benefits, wellness benefits including an employee assistance program (EAP), paid time off and paid company holidays, and family care and bonding leave.
Interns are eligible for some of the benefits listed. Our pay ranges are determined by role, level, and location. The range displayed reflects the pay for this position in the primary location identified in this posting. Actual pay depends on several factors, including state minimum pay wage rates, location, job-related skills, experience, and relevant education level or training. We are committed to complying with all applicable federal and state minimum wage requirements where applicable. If applicable, your recruiter can share more about the specific pay range for your preferred location during the hiring process.
Use of AI Statement
At KLA, our interviews seek to understand your individual skills, problem-solving approach and authentic thinking. To ensure a fair and consistent evaluation, the use of AI, recording tools or other technologies to generate, suggest or provide responses during interviews—whether virtual or in person—is not permitted unless explicitly approved in advance as part of a reasonable accommodation or invited by the interviewer. Use of these tools may interfere with our ability to evaluate your individual qualifications and affect your candidacy. KLA is committed to advancing innovation through responsible AI, and we value candidates who share this mindset.
Equal Opportunity Statement
KLA is proud to be an Equal Opportunity Employer. We will ensure that qualified individuals with disabilities are provided reasonable accommodation to participate in the job application or interview process, to perform essential job functions, and to receive other benefits and privileges of employment. Please contact us at [email protected] or at +1-408-352-2808 to request accommodation.
For additional information, view the US Know Your Rights poster on the U.S. Equal Employment Opportunity Commission website.
Salary Context
This $105K-$180K range is in the lower quartile for AI Software Engineer roles in our dataset (median: $194K across 269 roles with salary data).
Role Details
About This Role
AI Software Engineers build the applications and systems that AI models run inside. They own the API layers, data pipelines, frontend integrations, and infrastructure that turn a model into a product users interact with. Every AI company needs engineers who can build the software around the AI.
The challenge is building reliable systems around inherently unreliable components. Models are probabilistic. They'll give different answers to the same question. They hallucinate. They're slow. They're expensive. Your job is to build an application layer that handles all of this gracefully while delivering a product that users trust and enjoy.
Across the 11,520 AI roles we're tracking, AI Software Engineer positions make up 5% of the market. At KLA, this role fits into their broader AI and engineering organization.
AI Software Engineer roles are among the most numerous in the AI job market. Every company deploying AI needs software engineers who understand AI integration patterns. The demand is broad, spanning startups to enterprises, across every industry adopting AI capabilities.
What the Work Looks Like
A typical week includes: building API endpoints that serve model inference with caching and fallback logic, designing the data pipeline that feeds context to a RAG system, implementing streaming responses in the frontend, debugging a race condition in the async inference pipeline, and optimizing database queries for the vector search layer. It's full-stack engineering with AI at the center.
AI Software Engineer roles are among the most numerous in the AI job market. Every company deploying AI needs software engineers who understand AI integration patterns. The demand is broad, spanning startups to enterprises, across every industry adopting AI capabilities.
Skills Required
Full-stack engineering skills with AI integration experience. Python and TypeScript are the most common requirements. You'll need to understand API design, database architecture, and how to build reliable systems around probabilistic outputs. Experience with streaming, async processing, and caching patterns is increasingly important as real-time AI applications proliferate.
Knowledge of vector databases, embedding APIs, and LLM integration patterns (function calling, structured outputs, retry logic) differentiates AI software engineers from general software engineers. Understanding cost optimization (caching strategies, model routing, batched inference) is valuable since inference costs can dominate application economics.
Strong postings describe the product you'll be building, the AI integration patterns you'll work with, and the scale requirements. Look for companies that have existing AI features and need engineers to improve and expand them, not companies that are 'planning to add AI' someday.
Compensation Benchmarks
AI Software Engineer roles pay a median of $220,000 based on 1,688 positions with disclosed compensation. Mid-level AI roles across all categories have a median of $193,200. This role's midpoint ($142K) sits 35% below the category median. Disclosed range: $105K to $180K.
Across all AI roles, the market median is $213,480. Top-quartile compensation starts at $264,000. The 90th percentile reaches $318,400. For comparison, the highest-paying categories include AI Safety ($266,050) and AI Engineering Manager ($253,461). By seniority level: Entry: $112,320; Mid: $193,200; Senior: $230,000; Director: $272,250; VP: $235,830.
KLA AI Hiring
KLA has 7 open AI roles right now. They're hiring across AI Software Engineer, Research Scientist, AI/ML Engineer. Positions span Ann Arbor, MI, US, Milpitas, CA, US, Ann Arbor, MI. Compensation range: $155K - $199K.
Location Context
Across all AI roles, 6% (708 positions) offer remote work, while 10,737 require on-site attendance. Top AI hiring metros: New York (3,257 roles, $221,000 median); San Francisco (2,766 roles, $260,550 median); Los Angeles (1,580 roles, $210,000 median).
Career Path
Common paths into AI Software Engineer roles include Software Engineer, Full-Stack Developer, Backend Engineer.
From here, career progression typically leads toward Staff Engineer, AI Architect, Engineering Manager.
If you're a software engineer, you're already 80% there. Learn the AI integration patterns: RAG, streaming inference, function calling, structured outputs. Build a project that demonstrates you can wrap an AI model in a production-quality application with proper error handling, caching, and user experience. That's the portfolio piece that gets you hired.
What to Expect in Interviews
Technical screens look like standard software engineering interviews with an AI twist. Expect system design questions about building reliable applications around probabilistic models: handling streaming responses, implementing retry logic for API failures, and designing caching strategies for LLM outputs. Coding rounds test standard algorithms plus practical integration patterns like async processing and rate limiting.
When evaluating opportunities: Strong postings describe the product you'll be building, the AI integration patterns you'll work with, and the scale requirements. Look for companies that have existing AI features and need engineers to improve and expand them, not companies that are 'planning to add AI' someday.
AI Hiring Overview
The AI job market has 11,520 open positions tracked in our dataset. By seniority: 140 entry-level, 5,630 mid-level, 4,176 senior, and 1,574 leadership roles (Director, VP, C-Level). Remote roles make up 6% of the market (708 positions). The remaining 10,737 roles require on-site or hybrid attendance.
The market median for AI roles is $213,480. Top-quartile compensation starts at $264,000. The 90th percentile reaches $318,400. Highest-paying categories: AI Safety ($266,050 median, 70 roles); AI Engineering Manager ($253,461 median, 58 roles); Research Engineer ($252,000 median, 476 roles).
AI Software Engineer roles are among the most numerous in the AI job market. Every company deploying AI needs software engineers who understand AI integration patterns. The demand is broad, spanning startups to enterprises, across every industry adopting AI capabilities.
The AI Job Market Today
The AI job market spans 11,520 open positions across 16 role categories. The largest categories by volume: AI/ML Engineer (8,769), AI Software Engineer (586), AI Product Manager (511). These three account for the majority of open positions, though smaller categories often have higher per-role compensation because of specialized skill requirements.
The seniority mix tells a story about where AI teams are in their maturity. Entry-level roles (140) are outnumbered by mid-level (5,630) and senior (4,176) positions, reflecting that most companies are past the 'build a team from scratch' phase and need experienced engineers who can ship production systems. Leadership roles (Director, VP, C-Level) total 1,574 positions, representing the bottleneck between technical execution and organizational strategy.
Remote work availability sits at 6% of all AI roles (708 positions), with 10,737 requiring on-site or hybrid attendance. The remote share has stabilized after the post-pandemic correction. Senior and specialized roles (Research Scientist, ML Architect) are more likely to be remote-eligible than entry-level positions, partly because experienced hires have more negotiating power and partly because these roles require less hands-on mentorship.
AI compensation is structured in clear tiers. The market median sits at $213,480. Top-quartile roles start at $264,000, and the 90th percentile reaches $318,400. These figures include base salary with disclosed compensation. Total compensation (including equity, bonuses, and sign-on) runs 20-40% higher at companies that offer those components.
Category matters for compensation. AI Safety roles lead at $266,050 median, while Prompt Engineer roles sit at $140,000. The spread between highest and lowest-paying categories reflects the premium on specialized technical skills versus broader analytical roles.
The most in-demand skills across all AI postings: Python (2,267 postings), Aws (1,178 postings), Azure (972 postings), Rag (924 postings), Gcp (715 postings), Pytorch (586 postings), Claude (572 postings), Kubernetes (559 postings). Python dominates, appearing in the vast majority of role descriptions regardless of category. Cloud platform experience (AWS, GCP, Azure) is the second most common requirement. The newer entrants to the top skills list (RAG, vector databases, LLM APIs) reflect the shift from traditional ML toward generative AI applications.
Frequently Asked Questions
Get Weekly AI Career Intelligence
Salary data, skills demand, and market signals from 16,000+ AI job postings. Every Monday.