Gpu Engineer • Mid-Level

Who you are

You have a BS degree in Computer Science, Electrical Engineering, or a related field, and possess strong skills in software and hardware performance analysis and optimization. With 3+ years of relevant industry experience, you are familiar with GPU programming models such as Metal and CUDA, and have experience with machine learning frameworks like PyTorch. Your background includes optimizing linear algebra algorithms and LLM inference for low latency at the implementation level.

You are adept at communicating across both hardware and software organizations, ensuring that your insights and analyses are understood and actionable. Your experience in optimizing LLM inference at scale in cloud or datacenter environments sets you apart, as you understand the intricacies of performance tuning in complex systems.

What you'll do

As a GPU Engineer at Apple, you will analyze the performance of linear algebra and machine learning frameworks, contributing to the Neural Accelerator effort. Your role will involve developing and optimizing GPU kernels to enhance the performance of machine learning applications. You will collaborate closely with cross-functional teams to ensure that the GPU architecture meets the demands of Apple's innovative products.

You will be responsible for conducting performance analysis and providing insights that drive architectural decisions. Your work will directly impact the efficiency and effectiveness of machine learning models deployed across Apple's hardware products. You will also engage in code reviews and mentor junior engineers, fostering a culture of excellence and continuous improvement within the team.

What we offer

Apple provides a collaborative and inclusive work environment where innovation thrives. You will have the opportunity to work on cutting-edge technologies and contribute to products that redefine industries. We encourage you to apply even if your experience doesn't match every requirement, as we value diverse perspectives and backgrounds.

In addition to competitive compensation, Apple offers comprehensive benefits and opportunities for professional growth. You will be part of a team that is dedicated to leaving the world better than we found it, making a meaningful impact through technology.