Gpu Ml Engineer • Mid-Level

Who you are

You have a technical BS/MS degree and at least 2 years of experience in GPU compute kernel framework development, maintenance, and optimization. Your background includes system-level programming and a solid understanding of computer architecture. You possess excellent programming and problem-solving skills, along with strong communication and collaboration abilities. A good grasp of machine learning fundamentals and a background in mathematics, particularly linear algebra and numerical methods, are also part of your skill set.

What you'll do

In this role, you will add optimized GPU compute kernels across various domains including Machine Learning, Image Processing, Linear Algebra, and Computer Vision. You will define and implement APIs in Metal Performance Shaders, ensuring robust compute solutions for accelerating machine learning networks on Apple Silicon. Your work will influence the design of compute and programming models in next-generation GPU architectures, contributing to the overall performance and capabilities of Apple's products.

What we offer

Apple is committed to fostering an inclusive and diverse workplace. You will be part of a team that values collaboration and innovation, working on cutting-edge technology that impacts millions of users. We encourage you to apply even if your experience doesn't match every requirement, as we believe diverse teams build better products. Join us in shaping the future of computing and machine learning at Apple.