Machine Learning Research Engineer, Input Experience - NLP

Apple • Cupertino, CA

2d

About The Position

Our team’s mission is to enhance ML powered user experiences on all Apple
platforms through personalized multimodal input, composition, and understanding,
with a global perspective. We achieve this daily by collaborating at the
intersection of natural language processing, machine learning, and software
engineering. We are responsible for the machine learning and software (non-UI)
development for several user-facing Apple Intelligence features, including
Writing Tools, Summarization, Found In Apps, and Messages/Mail Smart Replies.
Additionally, we oversee the keyboard machine learning and software stack, which
includes autocorrection, suggestions, and inline completions. If you’re
passionate about being part of an ambitious, organized, and collaborative team
that delivers user experiences with pioneering ML partnered with the best UI
designs, join the Input Experience NLP team. Here, you’ll have the opportunity
to transition from building groundbreaking NLP models to optimizing them for
various hardware backends and user interfaces that create an enchanting
experience.

DESCRIPTION

As a Machine Learning Research Engineer on our team, you’ll build and
iteratively refine model pipelines that enable machine learning driven user
experiences on Apple products. You’ll conduct experiments and create prototypes
for innovative approaches to enhance the quality of our models and expand their
intelligence. Additionally, you’ll implement the building blocks and
infrastructure that integrate these innovations into our production pipelines
and contribute evaluation metrics to measure progress.

Requirements

• MS or PhD in Computer Science or related field with at least 2 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

Nice To Haves

• 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, RHLF, prompt engineering, data synthesis, automatic evaluation, and RAG
• Excellent written and verbal communication skills
• History of developing in Python and / or Swift
• Record of publications, innovations, and/or leadership

Responsibilities

• build and iteratively refine model pipelines that enable machine learning driven user experiences on Apple products
• conduct experiments and create prototypes for innovative approaches to enhance the quality of our models and expand their intelligence
• implement the building blocks and infrastructure that integrate these innovations into our production pipelines
• contribute evaluation metrics to measure progress

What This Job Offers

• Job Type: Full-time
• Career Level: Mid Level