AIML - ML Platform Engineer (Apple Ecosystem), Data Intelligence Platform

We are looking for engineers with a passion for using machine learning to create intelligent applications. Be part of a highly accomplished, deeply technical and close-knit team of machine learning engineers, you will create tools that are used by millions of people. You will design and implement new machine learning algorithms and techniques, and collaborate with the most innovative product development teams in the world. This team builds the infrastructure that enables product teams across Apple to develop machine-learning solutions that power amazingly intelligent user experiences. We are looking for new energetic members to join the ML team. In this role, you will build the infrastructure, tools, and develop machine-learning solutions. You will also have the opportunity to engage with exciting new-product teams around Apple, and use your machine learning skills to solve challenging technical problems in our next-generation products that will delight millions of people.

Description
As a member of the AIML team, you will design, implement and ship a scalable, reliable, and easy-to-use machine learning platform and tools that will be used by Apple product teams. You will also collaborate with teams across Apple, who are building the new, compelling intelligent applications in the world. You bring a strong hands-on mentality that enables you to own engineering projects from inception to shipping product. You will also be a trusted advisor for best practice machine learning development.

Preferred Qualifications
Solid knowledge in LLM inference stacks, and hands-on experiences in latency optimization, e.g. attention/KV-cache optimization, GPU profiling, memory bandwidth analysis, multi-GPU scaling across diverse accelerator architectures etc
Hands-on iOS/macOS development experiences
A real passion for delivering secure and high-performance AI inference platforms at scale
Excellent problem-solving and debugging skills
Creative, collaborative & product-focused

Minimum Qualifications
Bachelor's degree or above in Computer Science or equivalent experience
10+ years of experience in industry-leading companies
Strong software development skills, with proficiency in Swift / C++ / Python
Experienced in building, deploying, and running Machine Learning applications or services
Experience working with a very large scale of data, familiar with data processing frameworks like Spark
Solid technical understanding in machine learning, deep learning, computer vision, natural language processing, or artificial intelligence
Strong written and verbal communication skills in both Chinese and English