onsite
GPU Compiler Engineer - Graphics, Game and ML - Apple
Software Engineer
Lead the design and implementation of GPU compiler technologies for Apple’s Metal framework, enabling high‑performance graphics, gaming, and machine‑learning workloads across iOS, macOS, and visionOS devices.
About the role
Key Responsibilities
- Design, implement, and optimize compiler passes for Metal and custom GPU languages, targeting Apple Silicon and future GPU architectures.
- Collaborate with hardware, graphics, and ML teams to define language extensions and transformation pipelines that expose low‑level GPU capabilities.
- Develop and maintain tooling for debugging, profiling, and performance analysis of GPU workloads.
- Contribute to open‑source LLVM components and internal compiler infrastructure, ensuring scalability and maintainability.
- Write technical documentation, conduct code reviews, and mentor junior engineers on compiler theory and GPU programming.
Requirements
- Strong background in compiler construction, with experience in LLVM or similar frameworks.
- Proficiency in C++ and Metal shading language, plus familiarity with GPU architecture and graphics pipelines.
- Experience optimizing code for performance, memory usage, and power efficiency on mobile and desktop GPUs.
- Knowledge of machine‑learning workloads on GPUs and related optimization techniques.
- Excellent problem‑solving skills, communication, and a collaborative mindset.