Apple logo

Machine Learning Research Engineer, Apple Intelligence, NLG

Apple
Full-time
On-site
Seattle, Washington, United States
Apple is where individual imaginations gather together, committing to the values that lead to great work. Every new product we build or service we create, we deliver is the result of us making each other’s ideas stronger. That happens because every one of us shares a belief that we can make something wonderful and share it with the world, changing lives for the better. It’s the diversity of our people and their thinking that inspires the innovation that runs through everything we do. When we bring everybody in, we can do the best work of our lives. Here, you’ll do more than join something — you’ll add something. Text generation is a key enabler for accelerated text input and intelligent interaction on Apple platforms. Our team is working on redefining user interaction with generative models for text generation. If you want to be part of an ambitious, organized and collaborative team that ships user experiences with pioneering ML partnered with best UI designs, come join the Input Experience NLP team in Software Engineering. The team’s work got highlighted in multiple WWDC keynotes including Intelligent Input in 2023, Writing Tools and Smart Reply in 2024. We exemplify Apple’s outstanding integration of hardware and software to create seamless input experiences. You will have the opportunity to go from building offline pioneering NLP models to optimization of the models for different hardware backends and user interfaces that make the experience magical. Our vision always includes a deep dedication to strengthening Apple’s privacy policy by achieving all of the above on device with powerful ML.

Description

As a key pillar of Apple Intelligence, input experience will be the main area where you bring impact to billions of users with your ML expertise, engineering passion, and programming skills. You will work with a hard-working and dedicated set of outstanding ML and software engineers on a wide range of most advanced text generation technologies such as context-augmented text rewriting, safety-controlled text composition, free-form text transformation, personalized smart interactions, etc. Our team has been working in this area for years and own the NLP and ML text input stack for the keyboard input that includes autocorrection, predictive typing on all Apple platforms. We also work on full stack ML applied to NLP and expose these key technologies across Apple on device and also to third party applications through the Natural-Language framework. If you want to amplify your strong ML and NLP skills into user experiences that will reach every person around you, this is the perfect opportunity!

Minimum Qualifications

  • Strong machine learning fundamentals
  • Strong programming and communication skills
  • Ph.D. in CS/EE/Physics/Statistics/etc. (or Bachelor/Master with 2 years of industry experience)

Key Qualifications

Preferred Qualifications

  • Actively exercising machine learning techniques, including implementing basic optimizers and regularizations; formulating and implementing loss functions for given tasks, applying insightful parameter tuning in model training and evaluation, reproducing and revising state-of-the-art research experiments; addressing accuracy and performance bottlenecks in distributed training and inference, and benchmarking different model compressing methods
  • Actively programming with high-quality codes across complex and large repositories, including implementing new optimizer or loss functions, algorithm optimizations for critical performance or accuracy, refactoring or redesigning of existing code-base for improved robustness and flexibility, processing large data set efficiently and accurately, establishing automated synthesis/training/evaluation pipeline across multiple sets of internal and external APIs, and applying common object-oriented design patterns
  • Familiar with common NLP algorithms and applications, including tokenization, language modeling, text decoding, text classifier etc
  • Experience of multi-modal modeling, presenting plans, progress, and results or demos regularly and concisely

Education & Experience

Additional Requirements

Pay & Benefits

  • Apple is an equal opportunity employer that is committed to inclusion and diversity. We take affirmative action to ensure 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.