KLA · 4 months ago
Lead AI Software and Systems Engineer
Ann Arbor, MI
Full-time
Onsite
Senior Level, Lead/Staff
$135K/yr - $229K/yr
7+ years expKLA is a global leader in diversified electronics for the semiconductor manufacturing ecosystem. They are seeking a highly skilled Lead AI Software and Systems Engineer to drive the development of next-generation intelligent software solutions, focusing on AI/ML knowledge and architectural leadership.
ElectronicsInformation TechnologyManufacturing
Responsibilities
Architect and Develop AI-Driven Software: Design and implement production-quality software that integrates LLMs and agentic AI frameworks to deliver groundbreaking capabilities to our customers
Code Quality and Software Design Leadership: With a focus on scalability and performance, help define and enforce best practices for clean, maintainable, and well-structured code
LLM Integration & Optimization: Fine-tune, prompt-engineer, and orchestrate LLMs for specific business use cases
Agentic AI Design: Build multi-step reasoning agents, tool-using agents, and autonomous workflows that interact with APIs, databases, and our existing codebases
Collaborate & Mentor: Partner with cross-functional teams (ML engineers, product managers, and designers) and mentor engineers on AI-driven development practices
Improve AI-Generated Code: Evaluate and refine LLM-generated code for correctness, performance, and maintainability. Develop feedback loops to enhance LLM coding capabilities
Stay Ahead of Trends: Research and adopt emerging technologies in AI, LLMs, and software engineering to maintain technical leadership
Qualification
Required
7+ years of professional software engineering experience, with at least 3+ years in AI/ML-driven systems
Proven expertise in Python, TypeScript/JavaScript, or similar languages for backend and AI integration
Strong understanding of LLMs, prompt engineering, and agentic AI frameworks to build modular AI software (e.g., DSPy, ell, LangChain, AutoGen)
Deep knowledge of software design principles, object-oriented programming and functional programming, system architecture, and code quality standards
Familiarity with MLOps, model fine-tuning, and evaluation techniques
Excellent problem-solving, communication, and leadership skills
Requires a minimum of 5 years of related experience with a Bachelors degree, or 3 years of experience and a Masters Degree
Benefits
Medical
Dental
Vision
Life
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
Paid company holidays
Family care and bonding leave
Company
KLA
Kla creates tools and services that promote innovation in the electronics industry.
10001+ employees
H1B Sponsorship
KLA has a track record of offering H1B sponsorships. Please note that this does not
guarantee sponsorship for this specific role. Below presents additional info for your
reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (343)
2024 (218)
2023 (191)
2022 (277)
2021 (200)
2020 (226)
Funding
Current Stage
Late StageLeadership Team
Lisa Grunwald
FISH Philosophy CEO
Rick Wallace
President & CEO
Recent News
news.com.au — Australia’s leading news site for latest headlines
2025-08-01
2025-08-01
Company data provided by crunchbase