A

Apple

Designs phones, computers and their operating systems

Senior Machine Learning Engineer

Natural Language Processing & LLMsConfirmed live in the last 24 hours
Full-timeBeijing, ChinaNo salary listedIn-person

About the Job

Requirements

  • At least 5 years of hands-on industry experience building, training, and shipping production-grade deep learning and natural language processing systems at scale.
  • A proven track record training and fine-tuning large language models, including supervised fine-tuning, direct preference optimization or reinforcement learning from human feedback, and parameter-efficient fine-tuning such as LoRA; strong understanding of model behavior, loss dynamics, and scaling laws.
  • Exceptional coding skills in Python and proficiency with modern deep learning frameworks, including PyTorch, DeepSpeed, Megatron-LM, Hugging Face, and vLLM.
  • Experience designing domain-specific datasets and synthetic data generation pipelines, building automated evaluation harnesses, and conducting systematic root-cause failure analysis.
  • Strong ownership, cross-functional communication, and project management skills, with the ability to navigate complex engineering constraints and deliver against tight product deadlines.

Responsibilities

  • Design, train, and fine-tune high-performance on-device large language models and modular adapter architectures, including LoRA, rank scaling, and routing, for specific Siri and Apple Intelligence tasks.
  • Develop and implement prompt and context compression, key-value cache optimization, tool-schema compaction, and efficient token representations to meet strict context-window limits.
  • Build and maintain rigorous offline and online evaluation pipelines; design benchmarks for agentic tool use, goal completion, and multi-turn conversational accuracy; conduct failure analysis to improve models.
  • Partner with platform engineers to integrate models and adapters into production builds, validate performance across over-the-air updates, and monitor runtime behavior at scale.
  • Lead the end-to-end lifecycle of on-device large language model adapters and sub-models, including training recipe formulation, data curation, architecture tuning, metric evaluation, error analysis, and on-device runtime optimization.
  • Work with Foundation Model and Product teams to scale Apple Intelligence across diverse languages and domains.

Desired Qualifications

  • Familiarity with high-performance C++ or Swift runtime integration.
  • Deep understanding of edge-AI constraints and deployment pipelines, including CoreML, ONNX, TensorRT-LLM, Apple Neural Engine, or Metal optimization.
  • Direct experience building agentic workflows, multi-turn reasoning, and function or tool-calling capabilities in compact or dense models in the 2B–9B parameter class.
  • A research track record demonstrated by publications in top-tier conferences such as NeurIPS, ICML, ICLR, ACL, or EMNLP, or a demonstrated history of shipping industry-leading generative AI products.

About the Company

Apple

Designs phones, computers and their operating systems

Apple designs and sells consumer electronics, from iPhone smartphones and Mac computers to tablets and wearable devices. Its operating systems manage apps and built-in functions for communication and work, while iCloud stores files and synchronizes data across devices. Customers buy additional digital services through subscriptions such as Apple Music and Apple TV, and developers distribute software through the App Store. Apple earns money from hardware sales, subscriptions and digital transactions, selling directly to customers and through retailers and mobile carriers around the world. It was founded in 1976 and has its headquarters in Cupertino, California.

Company size
10,001+
Company stage
IPO
Headquarters
Cupertino, CA
Founded
1976

Recently Posted Jobs

Explore more opportunities at Apple

See more jobs