Senior AI Infrastructure Engineer at KLA

Ann Arbor, Michigan, United States

Apply Now

Not SpecifiedCompensation

Senior (5 to 8 years), Expert & Leadership (9+ years)Experience Level

Full TimeJob Type

UnknownVisa

Semiconductor, ElectronicsIndustries

Requirements

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 to 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; or Bachelor's Level Degree and 5 years related work experience

Responsibilities

Architect and build the MLOps backbone of 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 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

Skills

Key technologies and capabilities for this role

MLOpsAIMachine LearningML DeploymentScalabilityReliabilityPerformance OptimizationSemiconductor ManufacturingDefect Detection

Questions & Answers

Common questions about this position

What qualifications and experience are required for the Senior AI Infrastructure Engineer role?

Candidates need a degree in Computer Science, Mathematics, or a related field, plus 7+ years of hands-on experience developing and deploying ML models for external customers. Deep expertise in Python and ML frameworks like PyTorch, TensorFlow, and Keras is required, along with proven experience with foundational models and modern AI toolchains.

What are the main responsibilities of this position?

Key duties include architecting and building the MLOps backbone, deploying ML models at customer sites, leading data pipelines and experimentation, collaborating cross-functionally, staying current with ML/AI research, and mentoring junior engineers.

What is the salary or compensation for this role?

This information is not specified in the job description.

Is this a remote position, or is there a location requirement?

This information is not specified in the job description.

What does KLA's company culture emphasize?

KLA fosters an exciting environment where expert teams of physicists, engineers, data scientists, and problem-solvers thrive on tackling hard problems, with a strong focus on innovation through 15% of sales invested in R&D, and encourages mentoring and continuous learning.

KLA

Provides process control and yield management solutions

About KLA

KLA provides process control and yield management solutions primarily for semiconductor manufacturers. The company offers advanced inspection tools, metrology systems, and computational analytics that help manufacturers identify and fix defects during production. This process enhances the quality and reliability of electronic devices, leading to higher production yields. KLA distinguishes itself from competitors by focusing on high-precision equipment and software that are essential for defect detection in semiconductor manufacturing. The company's goal is to promote sustainability by committing to using 100% renewable electricity in its operations by 2030.

Milpitas, CaliforniaHeadquarters

1975Year Founded

IPOCompany Stage

Industrial & Manufacturing, EnergyIndustries

5,001-10,000Employees