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

Open 33d

Referral available at Apple

An employee here can refer you. The referrer stays anonymous and reaches out to you directly if interested.

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.

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.

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

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
  • Expertise and hands-on experiences in Confidential Computing, e.g. hardware attestation, confidential VM, TEE, secure enclave 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