Job Details

Weekly Hours: 40

Role Number: 200629486-3337

Summary

Imagine what you could do here. At Apple, great ideas have a way of becoming great products, services, and customer experiences very quickly. Join us in enabling the next generation of intelligent customer experiences in Apple’s products and services with the latest advancements in Generative AI and Large Language Models! We are seeking a Senior ML Engineer to join our dynamic team. The ideal candidate has extensive experience with ML frameworks, GenAI lifecycle, TPU/GPU/CPU workflows, and is knowledgeable of the Python ML ecosystem. This is a hands-on individual contributor role ideal for an engineer who thrives in ambiguity, moves seamlessly between prototyping and production, and is excited to push the frontier of applied AI through practical, elegant engineering.

Description

As a Senior Machine Learning Engineer, you will:

-Design and build infrastructure to support AI features that empowers billions of Apple customers
-Develop tools used internally at Apple to build AI experiences through the entire lifecycle - from experimentation to production inference emphasizing observability, reproducibility, and modularity
-Work cross functionally with MLE, DS, Infra and others, staying on the cutting edge of tooling that supercharges ML/AI innovation at Apple
-Contribute to architectural direction for Apple’s broader AI ecosystem designing for flexibility across future projects

Minimum Qualifications

• 8+ years of experience in ML engineering, software engineering or applied AI roles

• Solid understanding of machine learning fundamentals, especially around large models, embeddings, and retrieval systems

• Proven experience building production-grade ML systems or intelligent data-driven products

• Proficiency in Python and advanced knowledge of the Python ML ecosystem

• Extensive experience building tools for ML lifecycle and launching ML-powered experiences to production

• Strong background in distributed systems, APIs, and scalable data/compute infrastructure

• Solid understanding of software engineering principles and design patterns

• Excellent communication skills and ability to work in a collaborative environment

• Experience with building on top of AWS or GCP cloud services

Preferred Qualifications

• Experience working with notebook ecosystems and web-based IDEs. Experience with building and deploy ML software in production to a wide range of infrastructure (CPU/GPU/TPU).

• Big data experience such as Apache Spark, Apache Ray

• Familiarity with MLOps tools (e.g., MLflow, Weights & Biases, Ray, Airflow)

• Some frontend development experience with Javascript / React

• Prior experience building an Notebook solution in a large organization

• Experience with building customer facing SDKs in Python

Apple is an equal opportunity employer that is committed to inclusion and diversity. We seek to promote equal opportunity for all applicants without regard to race, color, religion, sex, sexual orientation, gender identity, national origin, disability, Veteran status, or other legally protected characteristics. Learn more about your EEO rights as an applicant (https://www.eeoc.gov/sites/default/files/2023-06/22-088_EEOC_KnowYourRights6.12ScreenRdr.pdf) .