Apple - Cambridge, MA
posted about 2 months ago
Full-time
Cambridge, MA
Computer and Electronic Product Manufacturing
About the position
The AIML - On-Device Machine Learning group at Apple is focused on enhancing on-device machine learning experiences, with a key mission to profile, analyze, and optimize ML inference performance across various models and devices. This role involves scaling and extending a significant ML benchmarking service that supports a wide range of Apple devices, from wearables to Apple Silicon Macs, while providing opportunities to explore the latest research in on-device machine learning.
Responsibilities
- Scale and extend the ML benchmarking service used across Apple devices.
- Profile, analyze, and optimize ML inference performance.
- Engage in model visualization, efficient inference, model compression, and federated learning.
- Collaborate with partner teams to enhance ML capabilities across Apple products.
Requirements
- Proficiency in Python and/or C/C++ programming and software design skills.
- Understanding of performance modeling and profiling of computer systems.
- Solid ML fundamentals including data processing, training regimes, and evaluation.
- Excellent collaboration and communication skills.
Nice-to-haves
- Masters or PhDs in Computer Science or relevant disciplines.
- Experience with full-stack web development (e.g., Django) and front-end JavaScript frameworks (e.g., Vue, React, Angular).
- Back-end system skills including containers (Docker), cloud orchestration (Kubernetes), and database management (SQL, Postgres).
- Experience in computer system power profiling.
- Experience developing code in training frameworks (e.g., PyTorch, TensorFlow, JAX) and familiarity with Transformers, CNNs, or Stable Diffusion.
Benefits
- Equal opportunity employer committed to inclusion and diversity.
