Machine Learning Engineer, Apple Intelligence

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.

Product Engineering International, part of the esteemed Software Engineering Organization, is seeking an accomplished ML Researcher to spearhead the development of cutting-edge generative modeling pipelines for a diverse range of popular international languages.

Description
As a key pillar of Apple Intelligence, 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 specifically focuses on generative technologies for key International markets and ensure that experience is top class for the popular languages in these markets. 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!

In this pivotal role, you will be at the forefront of natural language processing, designing and implementing end-to-end solutions that will:

- Drive Innovation in Generative Modeling: Focus exclusively on building and optimizing generative models for international languages.

- Full-Stack Language Expertise: Own the entire generative modeling pipeline, from data preprocessing and model architecture design to training, evaluation, and deployment.

- Language Diversity: Prioritize working with a variety of popular international languages, ensuring our models are culturally and linguistically adaptable.

Are you a passionate ML researcher with a deep interest in natural language processing and a talent for building generative models that work seamlessly across languages? Join our team and make a meaningful impact!

Preferred Qualifications
Proven experience leveraging AI-assisted development tools - such as Claude, Copilot, or equivalent - to meaningfully accelerate experimentation, code development, and research workflows.
Full-stack expertise spanning the entire generative modeling pipeline - from data preprocessing and model architecture design through training, evaluation, and deployment - complemented by a solid foundation in software engineering principles, memory management, code architecture, feature comprehension, framework proficiency, and debugging.
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.
Experience of multi-modal modeling, presenting plans, progress, and results or demos regularly and concisely
Experience in NLP modeling for Indian languages is desired.
Strong problem solving and drive what matters
Results oriented with a desire to work in a fast-paced and fight for excellence.
Ability to communicate effectively and innovate to the details.

Minimum Qualifications
4+yrs of machine learning expertise.
Strong programming using OOPS/C/C++/Objective C/Swift in addition to Python
Good exposure to Deep Learning libraries like PyTorch and TensorFlow.
Thorough experience with common NLP algorithms and applications, including tokenisation, language modelling, text decoding, text classifier etc.,
Proven experience navigating and contributing to large-scale, complex codebases, with a disciplined focus on code quality, maintainability, and comprehensive test coverage.