AI/ML Engineering Intern at KLA

Milpitas, California, United States

Apply Now

Not SpecifiedCompensation

InternshipExperience Level

InternshipJob Type

UnknownVisa

Semiconductor, ElectronicsIndustries

Requirements

Currently pursuing MS or PhD in Computer Science, Electrical/Computer Engineering, Mathematics, Physics, or related field (exceptional BS candidates considered)
Hands-on experience training deep learning models for vision (classification, detection, segmentation, or anomaly)
Strong foundations (implied in AI/ML, computer vision, and related technical areas)

Responsibilities

Advance AI/ML models for defect detection, classification, and anomaly discovery on real wafer imagery; design experiments, define metrics, and drive SOTA baselines forward
Curate and engineer datasets (balancing/partitioning, augmentation strategies, labeling guidance) to maximize label efficiency and generalization across layers, technologies, and tools
Train, evaluate, and iterate using modern deep learning frameworks; build repeatable pipelines for preprocessing, training, inference, and reporting
Prototype algorithmic ideas (e.g., semi/self-supervised learning, domain adaptation, uncertainty estimation, active learning) and transition the best candidates into production-quality code
Optimize for scale and speed (efficient ops, quantization/pruning, GPU utilization) to meet on-tool and near-real-time constraints
Collaborate cross-functionally with system engineering and software teams on integration, testing, and demo/beta evaluations; contribute to documentation and quality practices

Skills

Key technologies and capabilities for this role

AIMLPythonSoftware EngineeringSemiconductorMetrologyInspection SystemsData Science

Questions & Answers

Common questions about this position

Is this position remote or on-site?

The position is on-site in Milpitas, CA.

What will I be working on in this AI/ML Engineering Intern role?

You will advance AI/ML models for defect detection, classification, and anomaly discovery on real wafer imagery, design experiments, define metrics, and drive state-of-the-art baselines forward, as well as curate and engineer datasets for label efficiency and generalization.

What skills are needed for this internship?

The role requires skills in AI/ML model development, particularly computer vision and deep learning for defect detection, along with dataset curation, engineering, augmentation strategies, experiment design, and metrics definition.

What is the company culture like at KLA?

Life at KLA is exciting with teams thriving on tackling hard problems, a strong focus on innovation, and investing 15% of sales into R&D, collaborating with expert teams of physicists, engineers, data scientists, and problem-solvers.

What are the details of this internship, like duration or start date?

It is a full-time internship with summer preferred, but the start date is flexible.

What is the compensation for this intern position?

This information is not specified in the job description.

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

Benefits

Health Insurance

Dental Insurance

Vision Insurance

Life Insurance

401(k) Retirement Plan

401(k) Company Match

Employee Stock Purchase Plan

Student Loan Assistance

Tuition Reimbursement

Wellness Program

Mental Health Support

Paid Vacation

Paid Holidays

Parental Leave

Risks

Emerging competition in solid-state battery technology may impact market share.

Rapid innovation in semiconductor processes may outpace KLA's current technology.

Potential delays in achieving renewable electricity goals could affect brand reputation.

Differentiation

KLA specializes in advanced inspection tools and metrology systems for semiconductors.

The company integrates computational analytics to enhance defect detection and yield management.

KLA is committed to sustainability, aiming for 100% renewable electricity by 2030.

Upsides

Rising demand for advanced inspection tools due to AI and IoT growth.

AI-driven predictive maintenance reduces downtime in semiconductor manufacturing.

Collaborations with cloud providers enhance data analytics and process optimization.

Land your dream remote job 3x faster with AI

Try Jobo Free