Software Engineer – AI/ML Intern

KLA

. Work on platforms for large-scale software systems .

Posted 11/2/2026full-timeUnited StatesEntry Level💰 $22 - $43 per hourWebsite

Core Competencies

Role fit

Core Competencies

Use this summary to align your resume positioning with the role.

Demonstrates expertise in developing machine learning models and data management solutions, with a strong foundation in multi-threaded software systems and proficiency in programming languages such as Python and C++. Collaborates effectively within cross-functional teams to deliver impactful solutions that align with business priorities.

Highest-signal resume keywords

Machine Learning Model DevelopmentPython ProgrammingMulti-Threaded Software SystemsTensorFlow or PyTorch FamiliarityData Structures and Algorithms Knowledge

ATS Keywords

Tailor your resume

Applicant Tracking System Keywords

Tip: use these terms in your resume and cover letter to boost ATS matches.

Hard Skills

PythonC++C#JavaMachine LearningDeep LearningData StructuresAlgorithmsMulti-ThreadingDistributed Systems

Soft Skills

Analytical CapabilitiesProblem-Solving SkillsGood Communication SkillsTeam PlayerMotivated by Team Success

Tools & Technologies

TensorFlowPyTorchNumPyScikit-learnCloud TechnologiesGPU ArchitecturesContainerizationCompute Clusters

Certifications & Qualifications

Masters Level DegreePhD Candidate

Industry Keywords

Data Management SolutionsMachine LearningDeep LearningDistributed Computing InfrastructureHigh-Impact Work

Tech Stack

Tools & technologies

CloudDistributed SystemsJavaNumpyPythonPyTorchScikit-LearnTensorflowC++

About the role

Key responsibilities & impact

Work on platforms for large-scale software systems
Contribute to data management solutions
Develop machine learning model training and inference solutions for KLA products
Deliver high-impact work advancing business priorities
Formulate creative solutions by analyzing complex data
Collaborate as a cross-functional team member with research scientists, software engineers, applications engineers, and process engineers

Requirements

What you’ll need

Masters level degree student or a PhD candidate
Hands-on development experience in Python, C++, C# or Java
Deep conceptual understanding of multi-threaded and multi-process software systems
Strong analytical capabilities and problem-solving skills
Familiarity with machine learning and deep learning solutions
Experience or knowledge of data structures and algorithms
Knowledge of traditional machine learning using Random Forest, XGBoost, and Logistic Regression
Knowledge of deep learning for regression, classification, and generative models
Familiarity with TensorFlow or PyTorch, NumPy, scikit-learn, and other ML/DL frameworks
Knowledge of distributed systems for data management
Knowledge of distributed computing infrastructure
Knowledge of cloud technologies for storage, containerization, and compute clusters
Knowledge of GPU architectures and data management
Good communication skills
Strong team player motivated by team success

Benefits

Comp & perks

Medical, dental, vision, life, and 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
Performance incentive programs may be available