GPU Engineer, Platform Architecture

Apple is a technology company known for innovative products, services, and customer experiences. They are committed to leaving the world better than they found it.
Machine Learning
Senior Software Engineer
In-Person
5,000+ Employees
7+ years of experience
AI · Enterprise SaaS

Description For GPU Engineer, Platform Architecture

Imagine what you could do here! At Apple, new ideas have a way of becoming extraordinary products, services, and customer experiences very quickly. Bring passion and dedication to your job and there's no telling what you could accomplish. Dynamic, inquisitive people and inspiring, innovative technologies are the norm here. The people who work here have reinvented entire industries with all Apple Hardware products. The same passion for innovation that goes into our products also applies to our practices strengthening our commitment to leave the world better than we found it!

The Platform Architecture GPU group is looking for a talented GPU Engineer with strong skills in performance analysis and development, across applications and operating system components.

As part of the team, a successful candidate will:

  • Analyze the performance of linear algebra and machine learning algorithms on Apple GPU platforms, pursuing investigations wherever they take you in Apple software
  • With our partner teams in Software Engineering and Hardware Technologies, formulate system-level strategies to address performance problems and unlock the next level of AI performance for our users
  • Help build the prototype software implementation, for simulated future hardware, then communicate what we learned to teams building production software for products

Minimum Qualifications:

  • BS degree
  • Experience with software and hardware performance analysis and optimization
  • Experience in GPU programming models such as Metal, CUDA, or similar

Preferred Qualifications:

  • MS or PhD in Computer Science, Electrical Engineering, or equivalent
  • 7 years of relevant industry experience
  • Experience with the internals of GPU drivers, compilers and/or accelerated libraries
  • Experience working specifically in CUDA C++ on ML and/or linear algebra algorithms
  • Experience with distributed algorithms for HPC (for example, supercomputing)
  • Experience communicating across both hardware and software organizations

Apple is an equal opportunity employer committed to inclusion and diversity. We take affirmative action to ensure 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.

Last updated 5 months ago

Responsibilities For GPU Engineer, Platform Architecture

  • Analyze the performance of linear algebra and machine learning algorithms on Apple GPU platforms
  • Formulate system-level strategies to address performance problems and unlock the next level of AI performance
  • Help build prototype software implementation for simulated future hardware
  • Communicate learnings to teams building production software for products

Requirements For GPU Engineer, Platform Architecture

  • BS degree
  • Experience with software and hardware performance analysis and optimization
  • Experience in GPU programming models such as Metal, CUDA, or similar

Interested in this job?

Jobs Related To Apple GPU Engineer, Platform Architecture

Multimodal Generative Modeling Research Engineer - SIML, ISE

Senior ML engineering role at Apple focusing on multimodal generative AI research and development for user-facing applications.

ISP Algorithm Engineer - Computer Vision

Senior Computer Vision Engineer role at Apple, focusing on developing and implementing advanced camera algorithms for Apple products, offering competitive compensation and benefits.

Computer Vision/Deep Learning Engineer

Senior Computer Vision/Deep Learning Engineer role at Apple, focusing on biometric ML algorithms and computer vision solutions for next-generation products.

ML Software Engineer

Senior ML Software Engineer role at Apple, building ML-inference applications and services on Apple Silicon, focusing on generative AI and cloud compute infrastructure.

Software QA Engineer, Machine Learning (SF)

Senior ML QA Engineer role at Apple, focusing on iOS and macOS creativity applications, requiring 5+ years of experience in QA and machine learning.