We use cookies to enhance your experience. By using our site, you agree to our use of cookies.

Apple Logo

Apple

GPU Compute Performance Engineer

Apply Now
🌎

Cambridge, Massachusetts, United States

1d ago
👀 8 views
📥 0 clicked apply

Job Description

Description

Description
The team is seeking extraordinary GPU and machine learning engineers who are passionate about providing robust compute solutions for accelerating machine learning workloads at both the system and GPU programming level on Apple Silicon. The ideal candidate will have a passion for squeezing the best performance possible out of our GPUs, and is able to explore the low level architectural details of the HW to achieve this. They will work closely with our GPU hardware architecture and design teams to help develop our GPU roadmap and to ensure Apple is building the right HW and SW features to make the best - and fastest - GPU products. This role’s responsibilities will include: * Working with internal partners to analyze and improve GPU and system performance of large scale ML deployments, such as Apple Intelligence, Private Cloud Compute, and MetalFX. * Working with internal and external partners to optimize their GPU based ML algorithm implementations, GPU compute applications, algorithms, and shaders to achieve the best possible performance on Apple platforms. * Working with internal hardware teams to define a hardware roadmap that continues to deliver best in class GPU performance as well as performance analysis capabilities, particularly in the areas of emerging GPU accelerated ML training and inference, GPGPU use-cases and workflows * Developing tools and frameworks to support internal and external developers with performance analysis on Apple Silicon GPUs.

Minimum Qualifications

Minimum Qualifications
  • Experience or interest in emerging GPGPU use cases in the areas of ML and compute
  • Experience or interest in optimizing compute workloads for GPU performance is a strong plus
  • GPU programming with Metal, DirectX, Vulkan, CUDA, Direct Compute, OpenGL, or OpenCL
  • Excellent software design and problem solving skills
  • Excellent system debugging skills
  • Excellent written and oral communication skills including the ability to communicate clearly and concisely across multiple audiences to explain analytical outcomes and technical roadblocks

Key Qualifications

Key Qualifications

Preferred Qualifications

Preferred Qualifications
  • Experience in ML frameworks such as pytorch, tensorflow, JAX and similar is highly desired
  • Experience in GPU compute kernel optimization experience for ML training and inference operations is highly desired

Education & Experience

Education & Experience

Additional Requirements

Additional Requirements

More

  • Apple is an equal opportunity employer that is 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. Learn more about your EEO rights as an applicant.

Apply Now

More Jobs at Apple

Locations
Jobs by Title
Jobs by Type