AI/ML Intern at KLA in Milpitas, CA

Summary: KLA is a global leader in diversified electronics for the semiconductor manufacturing ecosystem. They are seeking an AI/ML Intern to research and prototype anomaly detection methods, design proof-of-concept ML solutions, and analyze computer vision models under real-world constraints. Responsibilities: - Research and prototype anomaly/defect detection methods for high-resolution industrial vision data - Design and implement proof-of-concept ML solutions from research ideas - Train, evaluate, and analyze computer vision models under real-world data constraints - Iterate on models using quantitative metrics and qualitative failure analysis Required Qualifications: - Currently pursuing or recently completed a Master's or PhD in AI/ML, CS, EE, or related field - Working towards a MS or PhD in AI/ML, CS, EE, or related field Preferred Qualifications: - Hands-on experience implementing and modifying modern vision models (CNNs, Transformers) rather than only using pre-trained models - Demonstrated experience training at least one vision model end-to-end (data preparation → training → evaluation), documented via coursework, research, or projects - Strong Python skills; experience with PyTorch or TensorFlow - Experience with unsupervised or weakly supervised learning is a plus - Exposure to multimodal or vision-language models is a plus - Strong analytical and problem-solving skills Required Skills: AI/ML, Computer Vision, Python Important Skills: PyTorch, TensorFlow, CNNs, Transformers Nice-to-Have Skills: Analytical Skills, Problem-Solving Skills 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