← All roles

SW Optimization Engineer AI/ML

Apple· Hardware
Apply on company site
Hybrid (Cupertino, CA), Hybrid (Seattle, WA)$184.7k - $324.8kFull-timePosted 2w ago

About the role

Role focused on analyzing workloads to identify performance bottlenecks in hardware and software, optimizing AI/ML workloads on Apple Silicon (Neural Engine, GPU, CPU), and collaborating to inform architecture decisions and performance improvements across silicon and software teams.

Responsibilities

  • Conduct performance studies to inform and validate architecture decisions.
  • Create optimized implementations of machine learning workloads on Apple Silicon, including Neural Engine, GPU, and CPU.
  • Collaborate with system teams to create performance models of emerging AI/ML techniques and analyze system architecture trade-offs.
  • Work with software development tools teams to deliver performance analysis instruments and optimized libraries and frameworks for AI/ML applications.

Qualifications

  • Bachelor’s degree or equivalent job-related experience in Computer Engineering, Computer Science, or a related field.
  • Experience in software development for at least one of the following hardware IPs: AI/ML HW accelerators, GPUs processing units, image/video encoders, or similar.
  • Experience with AI/ML, graphics, or HPC performance benchmarks and workloads.
  • M.S. or Ph.D. in Computer Science, Computer Engineering, Electrical Engineering, or a closely related field.
  • 10+ years of relevant experience in software performance optimization, performance analysis tools, performance optimization process and development of efficient computational algorithms.
  • Knowledge of computer architecture fundamentals.
  • Proficiency in some of the C/C++ family programming languages, and scripting languages such as Python.
  • Ability to prototype and benchmark algorithms on CPU, GPU, and Neural Engine platforms, analyze performance metrics, and create high-level complexity models, or develop targeted highly efficient low-level performance libraries for AI/ML accelerators or GPUs.
  • Proficiency in popular AI/ML frameworks, such as PyTorch, and relevant software stacks.
  • Knowledge of operating system internals and compiler technologies.
  • Technical aptitude and curiosity, as well as ability to collaborate effectively with team members, partners, and stakeholders.

Skills

Required:Performance analysisAlgorithm benchmarking

Benefits

  • Comprehensive medical and dental coverage
  • Retirement benefits
  • Range of discounted products and free services
  • Reimbursement for certain educational expenses including tuition
  • Discretionary bonuses or commission payments and relocation (eligible in some cases)
  • Apple shareholder opportunities through stock programs
  • Employee stock purchase plan
  • Other benefits as described in Apple Benefits page
Last updated Oct 9, 2026. Interested? Apply on the company site.

About Apple

Discover the innovative world of Apple and shop everything iPhone, iPad, Apple Watch, Mac, and Apple TV, plus explore accessories, entertainment, and expert device support.

Discover the innovative world of Apple and shop everything iPhone, iPad, Apple Watch, Mac, and Apple TV, plus explore accessories, entertainment, and expert device support.

AirPods Max 2Apple ArcadeApple CardApple Fitness+Apple MusicApple TVApple TV+Apple Watch Series 11
Company profileWebsite

More open roles at Apple