AI/ML Engineering Intern – Model Evaluation,… at KLA · Milpitas

Summary: KLA is a global leader in diversified electronics for the semiconductor manufacturing ecosystem, focusing on innovation and R&D. The AI/ML Engineering Intern will work on ensuring AI models are accurate and scalable, assessing model performance, and collaborating with engineering teams on integration and deployment. Responsibilities: - Evaluate AI Models: Assess correctness, robustness, and performance using industry-standard metrics - Scalability Testing: Design and execute stress tests for large datasets and high-throughput inference environments - Extensibility Analysis: Explore modular architectures and frameworks to integrate new features without retraining from scratch - Assist in developing autonomous AI agents and multi-agent workflows - Build scripts and dashboards for benchmarking and monitoring - Collaborate with engineering teams on integration and deployment Required Qualifications: - Currently pursuing a BS, MS, or PhD in Computer Science, Data Science, or related field - Experience with Python, ML/DL frameworks (PyTorch/TensorFlow) - Familiarity with Agentic AI tools (LangChain, AutoGen) and LLM orchestration - Basic knowledge of MLOps and cloud platforms - Coursework or projects in AI agents, model evaluation, or workflow automation - Strong problem-solving and communication skills - Requires less than 1 year of related experience Required Skills: Python, ML/DL frameworks, Agentic AI tools Important Skills: MLOps, cloud platforms Nice-to-Have Skills: problem-solving, communication 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