AIML - Sr Machine Learning Engineer, Data and ML Innovation

Apple Inc.; Cupertino (CA); On-site; USD 150,400 - 277,600; Full time

Benefits offered by this job

Employee stock programs

Discretionary bonuses

Relocation assistance

Education reimbursement

Medical and dental coverage

Job summary

Apple Inc. is seeking a Senior ML Engineer to drive innovations for Foundation Models within the FM Data organization. You will contribute to model hill climbing, data augmentation strategies and training efficiency across Apple Intelligence features used by millions across devices.

This role requires strong ML modeling experience, advanced framework knowledge, and a passion for scalable software. You will collaborate with cross-functional teams to push the state-of-the-art in AI while ensuring

Qualifications

• 5+ years hands-on ML engineering experience.
• Master's or PhD in CS/EE/Math.
• Experience as ML modeler/scientist with ML and DL frameworks.
• Familiar with multi-modal data and large models.
• Strong software engineering skills and scalable code.

Responsibilities

• Algorithm development: prompts, responses and CoT signals.
• Model evaluation with balanced eval-sets and error analysis.
• Ablation experiments to test data augmentation and hyper-parameters.
• Data processing and filtering of large data sources.

Skills

5+ years ML engineering

Strong software engineering

Multimodal data experience

ML modeler/scientist experience

Education

Master's/PhD in CS/EE/Math

Tools

PyTorch

TensorFlow

JAX

Job description

AIML - Sr Machine Learning Engineer, Data and ML Innovation

Cupertino, California, United States Machine Learning and AI

Do you want to play a part in the next revolution in Foundation Models? Contribute to model hill climbing for Apple Intelligence features that leverage Apple Foundation Models, and work with the people who built the intelligent products that helps millions of people get things done — just by asking or typing! The vision for the AI/ML FM Data organization is to improve Foundation Models by leveraging data from a variety of sources: crawl, license, vendor and internal crowd-sourcing. As a Sr ML Engineering on the team, you will drive ML innovations, identify key opportunity areas where data can play a crucial role and experiment with various data augmentation strategies to improve model training efficiency and performance.

Description

We are looking for people with a track record in building models and model-driven products to affect user experiences. Join us, and impact hundreds of millions of customers across billions of their interactions with foundation model powered Apple Intelligence features, that are available on iPhone, iPad, HomePod, Mac, Watch, CarPlay, and tv across more than 30 languages.

• Algorithm development: Define signals that are important in prompts, responses and CoT reasoning steps. These usually require a fine-tuned model for specific use cases.
• Model evaluation: Understand the importance of a balanced eval-set. Ability to perform error analysis to figure out how to improve model capabilities.
• Ablation experiments: Test your data augmentation strategies via ablation experiments. Comfortable debugging training errors, and tune hyper-parameters and data mixture to achieve desired outcome.
• Data processing and data filtering: Ability to efficiently process and filter very large amounts of data, often times messy.

Minimum Qualifications

• 5+ years of hands on ML engineering experiences.
• Master or PhDs in Computer Science, Electric Engineering or Mathematics.
• Have prior experience as an ML modeler/scientist/researcher. Knowledgeable in classic machine learning algorithms (SVM, Random Forest, Naive Bayes, KNN etc), as well as comfortable with more modern deep learning frameworks (PyTorch, Tensorflow, Jax).
• Familiarity with multi-modal data and large models including image and video.
• Possess strong software engineering skills and mindset. Have a high bar for engineering code quality and scalability.

Preferred Qualifications

• Hands on experiences with different phases in LLM model training, including LoRA, SFT, RLHF, reward modeling.
• A good communicator with clear and concise, active listening and empathy skills.
• Are self-motivated and curious. Strive to continually learn on the job.
• Have demonstrated creative and critical thinking with an innate drive to improve how things work. Have a high tolerance for ambiguity.

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 $150,400 and $277,600, 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.

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.

At Apple, we believe accessibility is a fundamental human right. You’ll find that idea reflected in everything here — in our culture, our benefits and our digital tools. By welcoming as many perspectives as possible, we help you build a career where you feel like you belong.

Learn about accessibility in Apple’s workplace.

Learn about reasonable accommodations for job applicants.

Apple accepts applications to this posting on an ongoing basis.