Senior Machine Learning Research Engineer, Natural Language Generation (NLG), Apple Intelligence

Apple

3+ Years | Cupertino, California, United States (On Site) | Full Time | 4 months ago

Apply Now

Job Summary

On the Input Experience NLP team, we build language models for intelligent text input across Apple platforms, including keyboard auto-correction, Writing Tools, and Smart Reply. We are looking for a Senior Machine Learning Research Engineer to deliver scalable, multilingual NLP solutions. You will build and refine training and evaluation pipelines, driving focused iteration to enhance user experience. This role involves integrating the latest ML innovations and working on features reaching everyday users, collaborating with various Apple teams on human interfaces, user studies, internationalization, ML technologies, and system integration. You will build and iteratively refine model pipelines that enable multilingual text input experiences on Apple products, conduct experiments, create prototypes for new approaches to improve model quality, and implement building blocks and infrastructure for production pipelines.

Must Have

MS or PhD in Computer Science or related field with at least 3 years of industry experience
Strong Python programming skills, with experience developing production-quality Python modules
Solid background in machine learning, data science, natural language processing, or statistics

Good to Have

Experience building and maintaining model pipelines end-to-end
Ability to design and perform experiments that bring ML and NLP research ideas to production
Familiarity with LLMs, such as SFT, RLHF, prompt engineering, data synthesis, automatic evaluation, and RAG
Background in linguistics, fluency in multiple languages, or passion for scaling NLP features for global audiences
Excellent written and verbal communication skills
History of developing Python packages and supporting other users
Record of publications, innovations, and/or leadership

Perks & Benefits

Discretionary employee stock programs
Discretionary restricted stock unit awards
Discounted Apple stock purchase
Comprehensive medical and dental coverage
Retirement benefits
Range of discounted products and free services
Reimbursement for certain educational expenses

Job Description

On the Input Experience NLP team, we build the language models that underpin intelligent text input across Apple platforms, from keyboard auto correction to the Writing Tools and Smart Reply features announced at WWDC 2024. We believe that generative AI is an incredibly promising technology that can help people communicate effectively and express themselves clearly, and we have only just begun to incorporate this technology into our products. On our team, you will help build the future and have a voice in what shape it takes. We are looking for a Senior Machine Learning Research Engineer to help deliver scalable, multilingual NLP solutions that empower our users to use intelligent text input in their language of choice. You will build and refine the training and evaluation pipelines that define our slice of Apple Intelligence, driving the focused iteration that makes the user experience magical. You will join an ambitious, organized, and collaborative team in a unique position to integrate the latest innovations from the ML community and work on features that reach everyday users, including your family and friends. You’ll work closely with teams across Apple, collaborating on human interfaces, user studies, internationalization, ML technologies, system integration, and more.

Description

As a Machine Learning Research Engineer on our team, you will build and iteratively refine model pipelines that enable multilingual text input experiences on Apple products. You will conduct experiments and create prototypes for new approaches to improve the quality of our models add new dimensions to their intelligence, in consideration of specific linguistic requirements and design considerations. Finally, you will implement the building blocks and infrastructure that bring these innovations into our production pipelines, and contribute evaluate metrics for measuring forward progress. KEY RESPONSIBILITIES: - Development and maintenance of modeling pipelines that scale to multiple languages and production deployment - Definition of robust automated evaluation metrics to facilitate hillclimbing model quality - Failure analysis to understand shortcomings of our models - Research into techniques for improving model behavior - Curation and synthesis of representative training and evaluation data - Implementation of experiments and simulations to assess the value of model changes - Collaboration with language experts and QA to refine modeling approach in consideration of language-specific requirements

MS or PhD in Computer Science or related field with at least 3 years of industry experience
Strong Python programming skills, with experience developing production-quality Python modules
Solid background in machine learning, data science, natural language processing, or statistics

Experience building and maintaining model pipelines end-to-end, from data curation to evaluation
Ability to design and perform experiments that bring ML and NLP research ideas to production
Familiarity with LLMs, such as SFT, RLHF, prompt engineering, data synthesis, automatic evaluation, and RAG
Background in linguistics, fluency in multiple languages, or a passion for scaling NLP features for global audiences
Excellent written and verbal communication skills
History of developing Python packages and supporting other users
Record of publications, innovations, and/or leadership

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 $147,400 and $272,100, 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.

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.

9 Skills Required For This Role

Communication Game Texts Quality Control Html User Experience Ux Prototyping Data Science Python Machine Learning

Similar Jobs

Research Development

AI Adoption Unit Lead

jetbrains • Amsterdam, Noord-Holland, Netherlands (Remote)