Machine Learning Research Engineer , Text Generation, Input Experience

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, what we deliver is the result of us making each other's ideas stronger. 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 the best UI designs, come join the Input Experience NLP team. Our work has been 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 Machine Learning.

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 Machine Learning 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 auto correction, 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 Machine Learning and NLP skills into user experiences that will reach every person around you, this is the perfect opportunity!

Preferred Qualifications

Familiar with model compression algorithms including quantization, pruning, distillations, and experience optimizing large diffusion models or language models

Experience with hardware architecture, software & hardware co-design

Experience with deploying large ML models in real world products

Actively programming with high-quality codes across complex and large repositories

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

Minimum Qualifications

Strong machine learning fundamentals

Knowledge of ML techniques such as implementing basic optimizers, applying parameter tuning in model training and evaluation, and reproducing research experiments

Strong programming and communication skills

Ph. D. in CS/EE/Physics/Statistics/etc. (or Masters with 4 years of proven experience)

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 $142,300 and $263,300, 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.