AIML - Sr. ML Engineer, NL Response Generation - Answers, Knowledge, & Information (AKI)

The Answers, Knowledge & Information team is revolutionizing the way hundreds of millions of people access information on their devices, all while keeping user privacy at the forefront. As an Applied ML team, we're pushing the boundaries of Apple Intelligence, result ranking, and innovative search technologies, all while running a low latency production service. Our work fuels intuitive information experiences across some of Apple's most iconic products, including Siri, Spotlight, Safari, Messages, Lookup, and more. Join us in shaping the future of how the world connects with information! We are looking for a senior Machine Learning Engineer with a passion for using machine learning to build intelligent search applications. Our team researches and implements novel query understanding, ranking and response generation techniques, machine learning algorithms and models that power amazing Search experiences across Apple products.

This role will have the chance to work closely on Natural Language response generation models, make improvements and perform scalable evaluation that will impact Apple Users and broader AIML community. Role responsibilities include:

• Contribute on research, design, implementation and evaluation of algorithms and models to enhance the quality and performance of personalized response generation models.
• Use advanced NLP, deep learning and LLM techniques to generate grounded responses for user intent.
• Analyze loss patterns in the current search and assistant stack and come up with insights, algorithms and techniques to resolve quality gaps with the goal of improving the top-line product metrics.
• Develop a long-term technical vision; propose a roadmap for team setting clear objectives.
• Collaborate with teams across the company to define product requirements and prioritize ranking criteria that optimize user satisfaction.
• Establish metrics and continuously improve performance of models by applying innovative ML techniques.

• 4+ years Experience in Machine Learning, NLP and applying these techniques at scale
• Strong software engineering skills in mainstream programming languages, such as: Python, Go, C/C++
• Strong communication skills
• Bachelors in Computer Science and industry work experience

• In-depth knowledge and expertise in applying Deep learning models, Large Language Models and their evaluations
• Extensive experience in building production quality systems or applications in search, recommendation systems, or information retrieval
• Experience using ML frameworks (pyTorch, JAX, TensorFlow, XGBoost etc.)
• Ability to quickly prototype ideas / solutions, and perform critical analysis
• Background in: personalization, user behavior modeling, and data-driven decision-making
• Advance degree (Master’s or Ph.D.) in Computer Science, Statistics, or related field, or equivalent industry work experience

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.