KLA · 20 hours ago
Senior AI Infrastructure Engineer
Ann Arbor, MI
Full-time
Onsite
Senior Level
$112K/yr - $190K/yr
7+ years expKLA is a global leader in diversified electronics for the semiconductor manufacturing ecosystem. The Senior AI Infrastructure Engineer will lead the development of modern AI models to detect nanoscale defects, collaborating with multi-functional teams to drive innovation in semiconductor technology.
ElectronicsInformation TechnologyManufacturing
Hiring Manager
Keziah Catherene Abraham
Responsibilities
Architect and build the MLOps backbone of our AI products, ensuring scalability, reliability, and performance
Collaborate multi-functionally with team members, designers, and engineers to define and implement technical requirements
Deploy sophisticated ML models at customer sites, ensuring seamless integration and high performance in production air-gapped environments
Stay ahead of the curve by integrating the latest ML/AI research into real-world applications
Lead the data pipeline—from dataset selection and preprocessing to quality assurance and model training
Lead experimentation and statistical analysis to fine-tune models for optimal accuracy and efficiency
Mentor and inspire junior engineers, fostering a culture of technical excellence and continuous learning
Qualification
Required
A degree in Computer Science, Mathematics, or a related field
7+ years of hands-on experience developing and deploying ML models for external customers—not just internal tools
Deep expertise in Python and ML frameworks like PyTorch, TensorFlow, and Keras
Proven experience with foundational models and modern AI toolchains
Strong analytical skills and a track record of solving sophisticated technical problems
Excellent communication skills—you can explain sophisticated concepts to both technical and non-technical audiences
Demonstrated leadership in managing projects or teams, with the ability to juggle multiple priorities in a fast-paced environment
Master's Level Degree and 3 years related work experience; Bachelor's Level Degree and related work experience of 5 years
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