ML Framework (MetalLM) Engineer

1 Day ago • All levels • $143,100 PA - $264,200 PA

Job Summary

Job Description

The team is looking for exceptional machine learning and GPU programming engineers to develop compute solutions for accelerating machine learning libraries on Apple Silicon. This role provides an opportunity to influence the design of compute and programming models in future GPU architectures. Responsibilities include optimizing code for efficient and scalable ML inference using distributed compute strategies, developing kernel and compiler level optimizations, applying advanced model optimization techniques, collaborating with various teams to align software performance, analyzing and improving performance metrics, and implementing Metal device backend features for ML training acceleration technologies.

Job Details

Our team is seeking extraordinary machine learning and GPU programming engineers who are passionate about providing robust compute solutions for accelerating Machine learning libraries on Apple Silicon. Role has the opportunity to influence the design of compute and programming models in next generation GPU architectures. Responsibilities: Work on cutting-edge ML inference framework project and optimize code for efficient and scalable ML inference using distributed compute strategies such as data, tensor, pipeline and expert parallelism. Develop kernel and compiler level optimizations and perform in-depth analysis to ensure the best possible performance across Server hardware families. Apply advanced model optimization techniques including speculation, quantization, compression, and others to maximize throughput and minimize latency. Collaborate closely with hardware, compiler, and systems teams to align software performance with hardware capabilities. Analyze and improve performance metrics such as end-to-end latency, TTFT, TBOT, memory footprint, and compute efficiency. Implement features of Metal device backend for ML training acceleration technologies If this sounds of interest, we would love to hear from you!

Similar Jobs

Looks like we're out of matches

Set up an alert and we'll send you similar jobs the moment they appear!

Similar Skill Jobs

Looks like we're out of matches

Set up an alert and we'll send you similar jobs the moment they appear!

Jobs in Cupertino, California, United States

Looks like we're out of matches

Set up an alert and we'll send you similar jobs the moment they appear!

Similar Category Jobs

Looks like we're out of matches

Set up an alert and we'll send you similar jobs the moment they appear!

About The Company

We’re a diverse collective of thinkers and doers, continually reimagining what’s possible to help us all do what we love in new ways. And the same innovation that goes into our products also applies to our practices — strengthening our commitment to leave the world better than we found it. This is where your work can make a difference in people’s lives. Including your own.


Apple is an equal opportunity employer that is committed to inclusion and diversity. Visit apple.com/careers to learn more.

Waukee, Iowa, United States (On-Site)

Cupertino, California, United States (On-Site)

Sunnyvale, California, United States (On-Site)

Cupertino, California, United States (On-Site)

Cupertino, California, United States (On-Site)

Cupertino, California, United States (On-Site)

Cupertino, California, United States (On-Site)

Cupertino, California, United States (On-Site)

Seattle, Washington, United States (On-Site)

Sunnyvale, California, United States (On-Site)

View All Jobs

Get notified when new jobs are added by Apple

Level Up Your Career in Game Development!

Transform Your Passion into Profession with Our Comprehensive Courses for Aspiring Game Developers.

Job Common Plug