Machine Learning Engineer - AI Evaluation & LLM Systems

Socket.dev; Cupertino (CA); On-site; USD 150,000 - 230,000; Full time

Job summary

Apple is seeking a Machine Learning Engineer to build systems that measure and improve AI experiences used by millions. You will develop ML models, evaluation frameworks, and scalable infra to understand model behavior and accelerate LLM and multimodal AI research.

Collaborate with researchers, software engineers, and product teams to translate cutting-edge research into production-ready solutions, analyze large-scale data, and benchmark AI quality.

Qualifications

• MS or PhD in CS, ML, EE or related field, or equivalent practical experience.
• 1–2 years of industry, academic or internship experience in ML/AI.
• Proficiency in Python; familiarity with C++ or another OO language.
• Experience with PyTorch, TensorFlow or JAX.
• Understanding ML fundamentals: supervised learning, model evaluation, statistics.
• Experience with data processing and model training via coursework or projects.
• Strong analytical, problem-solving, and communication skills.

Responsibilities

• Build systems that measure and improve AI experiences used by millions of people.
• Develop ML models, evaluation frameworks, and scalable infrastructure.
• Analyze large-scale datasets and benchmark model quality.
• Collaborate with researchers, software engineers and product teams to deploy production-ready solutions.

Skills

Python

C++

PyTorch

TensorFlow

JAX

ML fundamentals

Analytical skills

Communication skills

Team collaboration

Education

MS/PhD in Computer Science, ML, EE or related field

Tools

Git

CI/CD

Job description

DESCRIPTION

As a Machine Learning Engineer, you will build the systems that measure and improve the quality of AI experiences used by millions of people. You will develop machine learning models, evaluation frameworks, and scalable infrastructure that enable teams to understand model behavior, identify regressions, and accelerate the development of large language models and multimodal AI. Working closely with researchers, software engineers, and product teams, you will transform cutting-edge research into production-ready solutions, analyze large-scale datasets, and develop new approaches for benchmarking and improving AI quality. This is a unique opportunity to solve challenging technical problems at the intersection of machine learning, software engineering, and data, while helping shape the future of AI at Apple.

MINIMUM QUALIFICATIONS

MS, or PhD in Computer Science, Machine Learning, Electrical Engineering, or a related technical field, or equivalent practical experience. 1–2 years of industry experience, or equivalent academic or internship experience, developing machine learning or AI solutions. Proficiency in Python and familiarity with C++ or another object-oriented programming language. Experience with one or more machine learning frameworks such as PyTorch, TensorFlow, or JAX. Understanding of machine learning fundamentals, including supervised learning, model evaluation, and statistical analysis. Experience working with data processing, model training, or experimentation through coursework, research, internships, or industry projects. Strong analytical, problem-solving, and communication skills with the ability to collaborate effectively in a team environment.

PREFERRED QUALIFICATIONS

Experience with large language models (LLMs), multimodal AI, or generative AI through internships, research, or personal projects. Experience building software or machine learning projects using modern engineering practices (Git, testing, CI/CD). Familiarity with distributed computing, cloud platforms, or large-scale data processing. Publications, open-source contributions, or participation in machine learning competitions. MS or PhD specializing in Machine Learning, Artificial Intelligence, or a related field.