Senior Machine Learning Performance Engineer - Remote

About the Role

At Apple, we're on the cutting edge of delivering transformative experiences through Artificial Intelligence. If you're passionate about pushing the boundaries of AI and hardware optimization, we want you to join our team as a Senior Machine Learning Performance Engineer. This remote role offers a unique opportunity to contribute to technologies that directly impact the daily experience of Apple customers worldwide.

What You'll Do

• Develop novel ML inference optimization strategies to improve performance and power efficiency on the Apple Neural Engine (ANE).
• Analyze and identify performance bottlenecks across the full stack, including model architecture, compiler, runtime, and hardware.
• Collaborate with Apple AI/ML, software, and silicon teams to co-design next-generation ML models and inference techniques optimized for ANE.
• Build and maintain performance profiling tools, benchmarking frameworks, and analysis infrastructure.
• Drive performance characterization, modeling, and optimization for large-scale ML workloads such as LLMs, diffusion models, and computer vision models.

Requirements

• Strong understanding of machine learning inference workloads, including LLMs, diffusion models, or computer vision models.
• Deep knowledge of computer architecture, including memory hierarchy, parallelism, dataflow, and SIMD/vector processing.
• Hands-on experience optimizing machine learning inference on hardware accelerators (GPU, TPU, NPU, etc.).
• Experience analyzing system-level performance and identifying optimization opportunities across software and hardware stacks.
• Strong programming skills in Python and C/C++.

Nice to Have

• MS or PhD in Computer Science, Machine Learning, Computer Architecture, or a related field.
• Experience with ML system optimization, performance modeling, or architecture evaluation.
• Experience developing profiling tools, benchmarking frameworks, or performance analysis tools.

What We Offer

• Comprehensive medical and dental coverage.
• Retirement benefits and a range of discounted products and free services.
• Reimbursement for certain educational expenses, including tuition.
• Discretionary bonuses or commission payments.
• Relocation support to help you settle into your new role.