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

MLOps

Machine Learning

ML Deployment

Scalability

Reliability

Performance Optimization

Semiconductor Manufacturing

Defect Detection

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