Apple - Austin, TX
posted about 2 months ago
Full-time
Austin, TX
Computer and Electronic Product Manufacturing
About the position
The GPU Architect role at Apple involves analyzing and optimizing the performance of linear algebra and machine learning algorithms on Apple GPU platforms. The position requires collaboration with software engineering and hardware technology teams to develop system-level strategies that enhance AI performance. The successful candidate will also be responsible for building prototype software implementations and communicating findings to production software teams.
Responsibilities
- Analyze the performance of linear algebra and machine learning algorithms on Apple GPU platforms.
- Collaborate with Software Engineering and Hardware Technologies teams to formulate strategies for performance improvement.
- Build prototype software implementations for simulated future hardware.
- Communicate findings and insights to teams developing production software.
Requirements
- BS degree in Computer Science, Electrical Engineering, or equivalent.
- Experience with software and hardware performance analysis and optimization.
- Experience in GPU programming models such as Metal, CUDA, or similar.
Nice-to-haves
- MS or PhD in Computer Science, Electrical Engineering, or equivalent.
- 15 years of relevant industry experience.
- Experience with the internals of GPU drivers, compilers, and/or accelerated libraries.
- Experience working specifically in CUDA C++ on ML and/or linear algebra algorithms.
- Experience with distributed algorithms for HPC.
Benefits
- Equal opportunity employer committed to inclusion and diversity.
