Machine Learning Engineer

Apple is hiring a Machine Learning Engineer to work on the Machine Intelligence Neural Design (MIND) team, combining research on foundation models and perception with ML feature delivery.

Responsibilities

• Build end-to-end ML model training and evaluation pipelines
• Develop production ML software that integrates models into Apple’s software infrastructure
• Create end-to-end demos for ML solutions
• Represent the team in cross-functional discussions
• Contribute across the full ML development pipeline, including dataset approaches, model training, runtime inference integration, and on-device model optimizations
• Collaborate closely with ML researchers, SW/FW engineers, and Operation/Data engineers to advance research and production efforts
• Support sensing-modality work, including real-time sensing visualizations and model prediction

Requirements

• 3+ years minimum experience (or equivalent) in ML software engineering
• BS or MS (or PhD) in Computer Science, Computer Engineering, or relevant fields
• Proficient programming skills in Python and one deep learning toolkit: JAX, PyTorch, or TensorFlow
• Experience with on-device ML model deployment and on-device optimization
• Experience working with multimodal data including text, images, audio, and other sensors
• Experience in developing production software
• Proficient in Swift, Objective-C, C++, or Go
• Strong foundation in machine learning (classical ML models, DNNs, and multimodal foundation models)
• Experience building Python/PyTorch model training and evaluation pipelines
• Experience prototyping and developing software applications (preferably in Swift)
• Strong communication and presentation skills
• Ability to work in a collaborative environment

Technologies

• Python
• JAX
• PyTorch
• TensorFlow
• Swift
• Objective-C
• C++
• Go

Location: Pittsburgh, PA (onsite)