Software Engineer, Generative AI & ML

Apple Inc.; Cupertino (CA); Hybrid; USD 180,000 - 230,000; Full time

Job summary

Apple Inc. is seeking a Senior AI/ML engineer to contribute hands-on to a Gen AI Solutions team. You will develop and fine-tune generative AI features, and design grounding strategies for data-relevant retrieval to support a multi-modal, multi-agent platform.

You will work with engineers, data scientists, and researchers to push state-of-the-art models, optimize latency and cost, and help roll out AI features across markets, including Greater China with Chinese language support.

Qualifications

• 5+ years hands-on experience in ML, backend engineering, data engineering.

Responsibilities

• Contribute to model development and fine-tuning workflows for generative AI features.
• Design and evaluate retrieval strategies for grounding large models in data.
• Prototype and benchmark multi-round agentic solutions for knowledge-intensive tasks.
• Partner with data and platform engineers for scalable deployment and monitoring.
• Stay current with advancements in large language modeling, vector search, RAG, and LLM Agents.
• Participate in architecture discussions with engineers and partner teams.
• Collaborate with product and business stakeholders to incorporate customer experience requirements into AI solutions.
• Support localization and rollout of AI features across multiple markets.

Skills

ML

Backend engineering

Data engineering

LLMs

RAG

Vector retrieval

Rust

Python

PyTorch

Tokenization

Education

Bachelor's or Master's in CS/ML

Equivalent practical experience

Tools

PyTorch

TensorFlow

JAX

Rust

Python

Vector search

Job description

Imagine what you could do here. At Apple, new ideas have a way of becoming great products, services, and customer experiences very quickly. Bring passion and dedication to your job and there's no telling what you could accomplish.The Customer Systems - Gen AI Solutions team is responsible for building multi-turn, conversational, agentic LLM-based applications and frameworks to support Apple customers across numerous lines of business. You’ll be contributing hands on to a team that consists of engineers, data scientists & researchers to enhance a multi-modal, multi-agent platform with a key focus on incorporating the latest models and advancements to improve, latency, cost and customer experience. This is an incredible opportunity to contribute innovation & research to a well established generative AI platform within Apple. There is a huge amount of opportunity and growth within this space!

Description

Contribute to model development and fine-tuning workflows for generative AI features.Design and evaluate retrieval strategies for grounding large models in product-relevant data.Prototype and benchmark multi-round agentic solutions for conversational knowledge-intensive tasks.Partner with data and platform engineers to ensure scalable deployment and monitoring.Stay current with advancements in large language modeling, vector search, RAG, and LLM Agents.Participate in architecture discussions with engineers and partner teams.Collaborate with product and business stakeholders to incorporate customer experience requirements into AI solutions.Support localization and rollout of AI features across multiple markets.

Minimum Qualifications

• 5+ years of hands-on experience in ML, backend engineering, data engineering
• 1-2 years of hands-on experience in training, fine-tuning, or evaluating LLMs
• Foundational understanding of RAG architectures and vector-based retrieval systems
• Exposure to multi-agent orchestration frameworks in Rust and Python.
• Familiarity with modern deep learning frameworks such as PyTorch, TensorFlow, or JAX.
• Experience with data preprocessing, tokenization, and pipeline automation.
• Proficiency in machine learning libraries (transformers, datasets).
• Strong problem-solving and collaboration skills, with the ability to learn quickly and adapt to production-grade systems.
• Experience working with Multi-modal LLMs to enable Voice capabilities is a plus or prior experience with STT, TTS systems.
• Experience with deploying to cloud environments (AWS, GCP, on-remote hybrid) is required.
• Experience in training, fine-tuning, or evaluating LLMs in Chinese for Greater China market
• Fluency in written and spoken Chinese to support the Greater China market

Preferred Qualifications

• Strong experience partnering with business and engineering team to deliver AI solutions.
• Bachelor's or Master's degree in Computer Science, Machine Learning, or related field, or equivalent practical experience.

Apple is an equal opportunity employer that is committed to inclusion and diversity. Apple provides reasonable accommodations to applicants with disabilities and in accordance with local requirements. Apple is a drug-free workplace.

At Apple, we believe accessibility is a fundamental human right. You’ll find that idea reflected in everything here — in our culture, our benefits and our digital tools. By welcoming as many perspectives as possible, we help you build a career where you feel like you belong.

Learn about accessibility in Apple’s workplace