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

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

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

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 .

Pay & Benefits

At Apple, base pay is one part of our total compensation package and is determined within a range. This provides the opportunity to progress as you grow and develop within a role. The base pay range for this role is between $181,100 and $318,400, and your base pay will depend on your skills, qualifications, experience, and location.

Apple employees also have the opportunity to become an Apple shareholder through participation in Apple's discretionary employee stock programs. Apple employees are eligible for discretionary restricted stock unit awards, and can purchase Apple stock at a discount if voluntarily participating in Apple's Employee Stock Purchase Plan. You'll also receive benefits including: Comprehensive medical and dental coverage, retirement benefits, a range of discounted products and free services, and for formal education related to advancing your career at Apple, reimbursement for certain educational expenses - including tuition. Additionally, this role might be eligible for discretionary bonuses or commission payments as well as relocation. Learn more about Apple Benefits.

Note: Apple benefit, compensation and employee stock programs are subject to eligibility requirements and other terms of the applicable plan or program.