ML Software Research Engineer

Socket.dev; Pittsburgh (Allegheny County); On-site; USD 140,000 - 200,000; Full time

Job summary

Apple's Machine Intelligence Neural Design (MIND) team seeks a self-driven ML Engineer to advance on-device ML features. You will own end-to-end development from data curation to model training, runtime inference, and device-specific optimizations for Apple devices.

Collaborating with ML researchers, SW/FW engineers, and operations/data teams, you will ship production-ready pipelines and on-device deployments, exploring multimodal data and novel architectures to deliver impactful user

Qualifications

• BS or MS plus 3–5 years in ML software engineering or PhD in CS/CE
• Strong ML foundation across classical ML, DNN, multimodal foundation models
• Experience with Python/PyTorch; prototyping and Swift development

Responsibilities

• Build ML training pipelines and production-ready runtime inference infrastructure
• Collaborate with ML researchers, SW/FW engineers, and data engineers to advance production efforts

Skills

Python

PyTorch

Swift

Communication

Collaboration

Education

Bachelor's or Master in CS/CE

PhD (optional)

Tools

PyTorch

TensorFlow

Swift

Go

Job description

As a Machine Learning Engineer in the Machine Intelligence Neural Design (MIND) team, you’ll have the chance to be part of an ML innovation organization that creates a wide range of impactful Apple features. Within MIND, you’ll work on cutting-edge research in fields like Foundation Models and Perception, and develop end products that bring this research’s impact to Apple users. This role places a strong emphasis on shipping ML-based features and products. You’ll be involved in the entire end-to-end ML development pipeline, which includes crafting creative approaches to datasets, model training, runtime inference integration, and on-device model optimizations. Our ideal team member is fearless in trying new things and is willing to iterate on ideas. We value team members who can quickly prototype and iterate, ultimately leading to high-quality implementations.

Description

Our team seeks self-driven machine learning software engineer with strong experience in building ML training pipeline and developing production quality run-time inference infrastructure. In this role you are expected to collaborate closely with ML researchers, SW/FW engineers, and Operation/Data engineers to advance different research and production efforts. Your role is to help deliver the needed pipeline for model development and evaluation, as well as, build the production software to integrated these models and related functionality within Apple’s software infrastructure. Having experience with Sensing Modality and building realtime sensing visualizations and model prediction.

Minimum Qualifications

• BS or MS + 3-5 years of ML SW Engineering or PhD in Computer Science, Computer Engineering or Relevant Fields
• Strong foundation in Machine Learning (classical ML models, DNN, and multimodal Foundation Models)
• Experience in building model training/eval pipelines in Python/Pytorch.
• Experience in prototyping and developing software applications (Preferably in Swift).
• Strong communication and presentation Skills
• Ability to work in a collaborative environment

Preferred Qualifications

• Proficient programming skills in Python and one of the deep learning toolkits such as JAX, PyTorch, or Tensorflow
• Experience in On-Device ML model deployment and on-device optimization
• Experience handling multimodal data including text, images, audio, and other sensors
• Experience in developing production software
• Proficient in Swift, Objective C, C++, or Go