On-Device ML Engineer: Foundation Models

Apple Inc.; Pittsburgh (Allegheny County); On-site; USD 130,000 - 200,000; Full time

Job summary

Apple Inc. in Pittsburgh, PA, seeks a Machine Learning Engineer to develop end-to-end ML pipelines and production-ready inference software within the MIND team. You will collaborate with ML researchers, SWE/FW engineers, and operations/data teams to deliver scalable model development and on-device optimizations.

You will build real-time demos and visualizations of sensing data streams and model predictions, with a focus on shippingML-based features and products across Apple software platforms.

Qualifications

• PhD in CS/CE or BS/MS with 3–5 years ML engineering exp.
• Strong foundation in ML with focus on LLM and multimodal models.
• Experience building model training/eval pipelines in Python/PyTorch.
• Experience prototyping and developing software (Swift preferred).

Responsibilities

• Build ML models and end-to-end training and evaluation pipeline.
• Develop ML production software.
• Build end-to-end demos for ML solutions.
• Represent the team in cross-functional discussions.

Skills

Python

PyTorch

Swift

ML Pipelines

On-device ML

Education

PhD in CS/CE

BS/MS + 3-5 yrs ML

Tools

PyTorch

TensorFlow

JAX

Job description

Apple Inc. in Pittsburgh, PA, seeks a Machine Learning Engineer to develop end-to-end ML pipelines and production-ready inference software within the MIND team. You will collaborate with ML researchers, SWE/FW engineers, and operations/data teams to deliver scalable model development and on-device optimizations.

You will build real-time demos and visualizations of sensing data streams and model predictions, with a focus on shippingML-based features and products across Apple software platforms.