AIML - Sr Machine Learning Performance Engineer, Siri and Information Intelligence

$175,800 - $312,200/Yr

Apple - Cupertino, CA

Full-time - Mid Level

Cupertino, CA

Computer and Electronic Product Manufacturing

About the position

The Machine Learning Engineer position at Apple focuses on optimizing the model inference stack for Siri, enhancing its performance and efficiency. This role involves collaboration with various product teams to identify the best foundation models for different use cases, ensuring that Siri meets its accuracy and performance targets. The engineer will play a critical role in diagnosing performance issues and developing innovative solutions to improve machine learning models used by Siri.

Responsibilities

• Analyze and optimize the performance of machine learning models and systems used by Siri.
• Develop and implement strategies for model tuning, parameter optimization, and efficient resource usage.
• Conduct performance benchmarking and develop tooling and metrics to measure model performance in terms of compute, memory, and latency.
• Collaborate with feature and product teams to consult on modeling decisions to achieve Siri performance objectives.
• Collaborate with hardware and software teams to integrate research findings into product implementation.

Requirements

• Strong understanding of Transformer and LLM architectures.
• Strong understanding of Operating System, Compiler, and Computer Architecture fundamentals.
• Expertise in optimizing software to take advantage of underlying hardware architecture.
• Experience in analyzing, identifying, and optimizing performance bottlenecks.

Nice-to-haves

• Expertise in optimizing model architectures for on-device inference.
• Experience working with modeling pipeline teams in model deployment and promotion pipelines.
• Creative, collaborative, and product-focused mindset.
• Excellent communication skills.

Benefits

• Comprehensive medical and dental coverage
• Retirement benefits
• Discounted products and free services
• Reimbursement for certain educational expenses including tuition
• Discretionary bonuses or commission payments
• Relocation assistance