Finding the best job has never been easier
Apple GPU Compute Performance Engineer United States, Massachusetts, Waltham 340811580
13.06.2024
Share
Description
This role’s responsibilities will include:* Working with internal partners to analyze and improve GPU and system performance of large scale ML deployments on single and multi SoC systems..* 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
- 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
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
Additional Requirements
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.
These jobs might be a good fit
