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 and implement ML solutions, and analyze computer vision models under real-world data 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 - 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 - Strong analytical and problem-solving skills Preferred Qualifications: - Experience with unsupervised or weakly supervised learning is a plus - Exposure to multimodal or vision-language models is a plus Required Skills: AI/ML, Computer Vision, Python Important Skills: PyTorch, TensorFlow, CNNs, Transformers Nice-to-Have Skills: Analytical Skills, Unsupervised Learning, Weakly Supervised Learning, Multimodal Models, Vision-Language Models, 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