Computational Support for Machine Learning and Transformers

The group is looking for an exceptional programmer to complement the team and make a difference. As a member of our fast paced group, you will have the unique opportunity to delight and inspire millions of Apple’s customers every day. You will work ina multi-functionall team which is implementing innovative and brand new routines to support the necessary computation for such things as vision algorithms, Machine Learning training and inference. You will push the state of the art in low level computation and drive them towards energy efficient and high performance implementations by tightly integrating software and hardware. The successful candidate will have excellent understanding and knowledge of Machine Learning primitives and micro architecture of NEON on ARM or AVX on Intel CPU cores from a vector programming perspective. Team members are engaged in the design and optimization of low level Computational Support for all aspects of Machine Learning and computational primitives in service of models such as Text to Speech, Voice Isolation, Apple Music Sing to mention a few. Being able to craft the fastest and the most energy efficient routines for a particular CPU core is a plus. Low level high performance programming experience is a must for this position. Being comfortable in vector assembly and low level C is a requirement. The ideal candidate would be at ease in developing both innovative and robust CPU core level algorithms derived from a particular technology’s need in a tight deadline.