Machine Learning Engineer • Mid-Level

Who you are

You have a strong systems background and a passion for building scalable, efficient, and production-grade solutions tailored for high-throughput GPU execution. Your experience includes working with machine learning frameworks and optimizing code for performance. You are familiar with advanced model optimization techniques such as speculation, quantization, and compression, which you apply to maximize throughput and minimize latency.

You possess deep knowledge of GPU programming and are excited about the opportunity to influence the design of compute and programming models in next-generation GPU architectures. Your collaborative spirit allows you to work closely with hardware, compiler, and systems teams to align efforts and achieve optimal performance across server hardware families.

Desirable

Experience with distributed compute strategies such as data, tensor, pipeline, and expert parallelism is a plus. Familiarity with custom-built server hardware and Apple silicon is also beneficial.

What you'll do

In this role, you will work on cutting-edge ML inference framework projects, optimizing code for efficient and scalable ML inference. You will develop kernel and compiler-level optimizations and perform in-depth analysis to ensure the best possible performance. Your responsibilities will include collaborating with various teams to align on project goals and deliver robust compute solutions for accelerating machine learning libraries on Apple Silicon.

You will have the opportunity to influence the design of compute and programming models in next-generation GPU architectures, ensuring that Apple remains at the forefront of machine learning technology. Your work will directly contribute to enabling Apple Intelligence through high-performance, distributed inference of GenAI applications on Private Cloud Compute.

What we offer

Apple provides a dynamic work environment where innovation thrives. You will be part of a team that is dedicated to pushing the boundaries of machine learning and GPU programming. We encourage you to apply even if your experience doesn't match every requirement, as we value diverse perspectives and backgrounds. Join us in shaping the future of technology at Apple.