Computational Support for Machine Learning and Transformers, Core OS

The Vector and Numerics Group within the Core Operating System is tasked with designing enhancing and improving various subsystems running on iOS macOS watchOs and tvOS. Most of this support is encapsulated in the Accelerate framework a well used library serving many technologies such as machine learning. The group is looking for an exceptional high performance 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 Apples customers every day.

You will work in a cross functional team which is implementing innovative and state of the art 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. nnTeam 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 technologys need in a tight deadline.

Excellent coding skills in ASM AND level algorithmic development in Machine Learning primitives BLAS and knowledge of vector Instruction Set Architectures (ISA) of ARM and in mathematics computer science or computer engineering

MS/PhD in computer science or computer engineering