AIML - Machine Learning Engineer, Search Query Understanding

Key Skills

C++, JAX framework, Large Language Models - LLMs, Natural Language Processing (NLP), PyTorch, TensorFlow

Job Description

We are looking for engineers with a passion for using machine learning to create intelligent search applications. In this highly accomplished, deeply technical and close-knit team machine learning specialists, software engineers and infra experts, you will create products that are used by millions of people. Our team researches new machine learning algorithms, models and techniques that will power amazing search experiences across Apple products. We are looking for new ambitious team members to join our ML group. You will have the opportunity to contribute to exciting projects around Apple, and use your data science, machine learning and analytical skills to solve challenging technical problems and ship novel products that will delight millions of people.

Key Qualifications

• Experience in Machine Learning, NLP, Large Language Models and applying these techniques at scale
• Strong coding and software engineering skills in a mainstream programming language, such as Python, Go, C/C++
• Familiarity with NLP/ML tools and packages like Jax, TensorFlow, pyTorch etc
• Practical experience building production quality applications related to natural language processing and machine learning
• In-depth knowledge of machine learning algorithms and ability to apply them in data driven natural language processing systems
• Ability to quickly prototype ideas / solutions, perform critical analysis, and use creative approaches for solving complex problems
• Ability to collaborate closely with multi-functional teams
• Clear oral and written communication skills

Description

As a member of the this team you have the opportunity to design and evaluate new algorithms, models and methods that will impact Apple and the broad ML community. Your responsibilities include:

 Researching, developing and implementing innovative language modeling techniques to further language understanding capabilities at Apple.

Building novel applications of LLMs to various search problems.

• Transferring novel machine learning solutions to improve various search and assistant product usecases.
• Implementing and launching end-to-end user facing improvements using performant and efficient ML models.
• Implementing tools, data pipelines and evaluation frameworks to enable development , iteration and launching ML ideas within the product scope.
• Analyzing the loss patterns in the current search and assistant stack and coming up with new features, algorithms and techniques to resolve those losses with the goal of improving the top-line product metrics.
• Collaborating with various product stakeholders across the company and partner with them to apply language understanding models to new product areas and usecases.
• Work on continuously improving the metrics and performance of models the team has launched in production by leverage state-of-the-art ML techniques.

Education & Experience

Bachelors in Computer Science or related field with a strong academic track record. PhD/Masters in Machine learning,

Additional Requirements

• NLP or related field would be given preference.
• 5+ years of industry experience working on Applied ML/NLP/Search Ranking problems.

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 $161,700 and $284,900, 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.