GPU Architect, 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
Principal Software Engineer
In-Person
5,000+ Employees
15+ years of experience
AI · Enterprise SaaS

Description For GPU Architect, Platform Architecture

Apple's Platform Architecture GPU group is seeking a talented GPU Architect with strong skills in performance analysis and development across applications and operating system components. The role involves analyzing the performance of linear algebra and machine learning algorithms on Apple GPU platforms, formulating system-level strategies to address performance problems and enhance AI performance, and building prototype software implementations for simulated future hardware. The ideal candidate will work closely with Software Engineering and Hardware Technologies teams to communicate findings and contribute to production software for Apple products.

Key responsibilities include:

  • Analyzing performance of linear algebra and ML algorithms on Apple GPU platforms
  • Developing system-level strategies to improve AI performance
  • Building prototype software for simulated future hardware
  • Collaborating with software and hardware teams

Required qualifications:

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

Preferred qualifications:

  • MS or PhD in Computer Science, Electrical Engineering, or equivalent
  • 15 years of relevant industry experience
  • Experience with GPU drivers, compilers, and accelerated libraries
  • Experience with CUDA C++ for ML and linear algebra algorithms
  • Knowledge of distributed algorithms for HPC
  • Strong communication skills across hardware and software organizations

Apple is an equal opportunity employer committed to inclusion and diversity, taking affirmative action to ensure equal opportunity for all applicants regardless of race, color, religion, sex, sexual orientation, gender identity, national origin, disability, Veteran status, or other legally protected characteristics.

Last updated 4 months ago

Responsibilities For GPU Architect, Platform Architecture

  • Analyze performance of linear algebra and ML algorithms on Apple GPU platforms
  • Formulate system-level strategies to address performance problems and enhance AI performance
  • Build prototype software implementations for simulated future hardware
  • Communicate findings to teams building production software for products

Requirements For GPU Architect, 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 Architect, Platform Architecture

Large Machine Learning Model Optimization Engineer

Lead the optimization and deployment of Apple's on-device Language and Vision AI models, focusing on LLM and diffusion model development.

AIML - Distinguished Engineer, Siri & Information Intelligence

Lead AI/ML initiatives at Apple as Distinguished Engineer for Siri & Information Intelligence, focusing on LLMs and reinforcement learning to enhance user experiences.

Real-Time Computer Vision Architect

Lead computer vision architecture role at Apple, focusing on real-time systems and ISP optimization, requiring 8+ years of experience and deep expertise in computer vision.

AIML - Principal Software Engineer (Production Engineering), Machine Learning Platform & Infrastructure

Principal Software Engineer role at Apple focusing on Machine Learning platform infrastructure and production engineering.

AIML - Lead Machine Learning Engineer, Global Safety, Siri and Information Intelligence

Lead Machine Learning Engineer position at Apple focusing on AI safety and generative models for Siri and Information Intelligence, offering competitive compensation and benefits.