ML Compute Acceleration Engineer

Technology company that designs and develops consumer electronics, software, and services.
$143,100 - $264,200
Machine Learning
Senior Software Engineer
In-Person
5+ years of experience
AI

Description For ML Compute Acceleration Engineer

Apple's Compute Frameworks team in GPU, Graphics and Displays org is seeking an exceptional ML Compute Acceleration Engineer to join their high-performance computing initiative. This role focuses on accelerating machine learning networks on Apple Silicon using GPU and Neural Engine technologies. You'll be working with cutting-edge technology, influencing next-generation GPU and Neural Engine architectures, and contributing to Apple's machine learning infrastructure.

The position offers a unique opportunity to work on crucial projects that power Apple Intelligence across iOS, macOS, and Apple TV platforms. You'll be responsible for optimizing computation graphs, implementing sophisticated APIs, and ensuring peak performance across Apple's hardware families. The role requires expertise in GPU computing, machine learning, and system-level programming.

As part of the team, you'll collaborate on developing GPU Compute acceleration technology, implementing Apple Intelligence features, and optimizing compute graphs across products. The ideal candidate combines strong technical skills in machine learning and GPU programming with excellent problem-solving abilities and communication skills.

This is an exceptional opportunity for someone passionate about high-performance computing and machine learning to make a significant impact on Apple's next-generation products. You'll be working with state-of-the-art technology while contributing to the foundation of Apple's machine learning capabilities. The role offers competitive compensation, comprehensive benefits, and the chance to work with some of the industry's best talents in a collaborative environment focused on innovation and excellence.

Last updated a minute ago

Responsibilities For ML Compute Acceleration Engineer

  • Adding optimizations in machine learning computation graph
  • Defining and implementing APIs in Metal Performance Shaders Graph
  • Developing and maintaining MLIR dialect in Apple and open source with upgrades using latest LLVM
  • Performing in-depth analysis, compiler and kernel level optimizations
  • Tune GPU and Neural Engine accelerated compute across products
  • Tuning the cost model and optimizing runtime dispatch to multiple IPs

Requirements For ML Compute Acceleration Engineer

Python
  • Proven programming and problem-solving skills
  • Good understanding of machine learning fundamentals
  • GPU compute programming models & optimization techniques
  • GPU compute framework development, maintenance, and optimization
  • Experience with system level programming and computer architecture
  • Background in mathematics, including linear algebra and numerical methods
  • Strong communication and collaboration skills
  • Strong background of building high performance, production quality software

Benefits For ML Compute Acceleration Engineer

Medical Insurance
Dental Insurance
Education Budget
Equity
Relocation Benefits
  • Comprehensive medical and dental coverage
  • Retirement benefits
  • Employee stock programs
  • Education reimbursement
  • Discretionary restricted stock unit awards
  • Employee Stock Purchase Plan
  • Discretionary bonuses
  • Relocation assistance

Interested in this job?

Jobs Related To Apple ML Compute Acceleration Engineer

Software QA Engineer, Machine Learning

Senior ML QA Engineer role at Apple, focusing on quality assurance for ML features in creativity applications, requiring 5+ years of experience and strong technical skills.

AIML - Sr Software Engineer, Siri on the Go

Senior Software Engineer role at Apple working on Siri's mobile experiences across car, Apple Watch, and AirPods platforms.

CAD Automation and ML Engineer

Senior CAD Automation and ML Engineer role at Apple, focusing on machine learning applications in circuit design automation and silicon development.

Camera and Photos Image Scientist

Senior imaging scientist role at Apple, developing next-generation camera technologies and image processing algorithms for Apple products.

AIML Sr SW Engineer - SystemRF

Senior AI/ML Engineer position at Apple, focusing on wireless systems optimization using machine learning and data analysis for product development.