Software Engineer – AI/ML Intern

KLA

. Develop platform capabilities 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 model training and inference solutions, with strong analytical capabilities and problem-solving skills. Proficient in Python, C++, C#, or Java, and familiar with cloud technologies and distributed systems for data management.

Highest-signal resume keywords

Machine Learning Model DevelopmentPython ProgrammingDeep Learning FrameworksData Management SolutionsMulti-Threaded Software Systems

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 StructuresAlgorithmsTensorFlowPyTorch

Soft Skills

Analytical SkillsProblem-SolvingCommunicationTeam PlayerMotivation by Team Success

Tools & Technologies

NumPyScikit-learnRandom ForestXGBoostLogistic RegressionDistributed ComputingCloud TechnologiesGPU Architectures

Industry Keywords

Software DevelopmentData ManagementMachine Learning SolutionsDeep Learning SolutionsDistributed Systems

Tech Stack

Tools & technologies

CloudDistributed SystemsJavaNumpyPythonPyTorchScikit-LearnTensorflowC++

About the role

Key responsibilities & impact

Develop platform capabilities for large-scale software systems
Work on data management solutions
Develop machine learning model training and inference solutions for KLA products
Deliver high-impact work supporting business priorities
Analyze complex data to formulate creative solutions
Collaborate as a strong team player across KLA’s software, research, applications, and process engineering teams

Requirements

What you’ll need

Masters level degree student or a PhD candidate
Exceptional hands-on software 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
Knowledge of data structures and algorithms
Experience with traditional machine learning, including Random Forest, XGBoost, and Logistic Regression
Experience with deep learning for regression, classification, and generative models
Experience 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 orientation and motivation 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