Senior Machine Learning Platform Engineer - AI, Search & Knowledge

Join us in building the AI, Search & Knowledge platform that enables teams at Apple to build Apple Intelligence and many other intelligent experiences across hardware, software and service products. As a Senior Search & Knowledge Platform Engineer, you'll create seamless integrations between our AI, Search & Knowledge platform and the diverse ML frameworks used across Apple, including PyTorch, JAX, and beyond.

Requirements

• Bachelor's degree in Computer Science, related field, or equivalent practical experience
• 10+ years of software engineering experience with strong backend development skills and platform engineering mindset
• Deep proficiency in Python with proven experience designing SDKs, libraries, and APIs for technical users
• Experience integrating with complex ML frameworks (PyTorch, TensorFlow, JAX, HuggingFace) and building production-grade backend services (REST/GraphQL APIs, microservices, databases)
• Track record of building end-to-end workflows that span multiple systems and teams, navigating complex technical landscapes to deliver pragmatic solutions
• Strong cross-functional collaboration and communication skills to understand diverse stakeholder needs, technical requirements, and articulate design decisions across ML engineering, infrastructure, and product teams
• Experience with cloud platforms (AWS, GCP, Azure) and container orchestration (Kubernetes)