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 Apple’s 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.

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.

Minimum Qualifications

• Excellent coding skills in ASM AND C.
• Low level algorithmic development in Machine Learning primitives, BLAS and FFT.
• Detailed knowledge of vector Instruction Set Architectures (ISA) of ARM and Intel.
• BS/MS in mathematics, computer science or computer engineering

Preferred Qualifications

• MS/PhD in computer science or computer engineering