AI/ML Engineering Intern – Model Evaluation, Scalability & Agentic AI jobs in United States
cer-icon
Apply on Employer Site
company-logo

KLA · 4 days ago

AI/ML Engineering Intern – Model Evaluation, Scalability & Agentic AI

positionMilpitas, CA
timeInternship
remoteOnsite
seniorityIntern
money$35/hr - $57/hr
date1+ years exp

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.

ElectronicsInformation TechnologyManufacturing
check
H1B Sponsor Likelynote

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

Qualification

PythonML/DL frameworksAgentic AI toolsMLOpsCloud platformsProblem-solvingCommunication skills

Required

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

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

Company

KLA

twittertwittertwitter
Glassdoor4.1
company-logo
Kla creates tools and services that promote innovation in the electronics industry.
dateFounded in 1975
location
Milpitas, California, USA
people-size10001+ employees
web-link
https://www.kla.com

H1B Sponsorship

KLA has a track record of offering H1B sponsorships. Please note that this does not guarantee sponsorship for this specific role. Below presents additional info for your reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (343)
2024 (218)
2023 (191)
2022 (277)
2021 (200)
2020 (226)

Funding

Current Stage
Late Stage

Leadership Team

leader-logo
Lisa Grunwald
FISH Philosophy CEO
linkedin
leader-logo
Rick Wallace
President & CEO
linkedin

Recent News

Business Live
US tech venture KLA secures multi-million-pound funding from Cardiff Capital Region to expand
2025-08-05
news.com.au — Australia’s leading news site for latest headlines
2025 Kia Tasman ute review
2025-08-01
thefly.com
KLA Corp. sees Q1 non-GAAP EPS $7.76-$9.30, consensus $8.15
2025-08-01
Company data provided by crunchbase