The Staff ML Infrastructure Engineer will join the Apple AI Platform team, focusing on building robust systems that connect scalable data pipelines with advanced machine learning workflows. This role involves shaping the technical direction of the platform and addressing complex system-level challenges. The team is dedicated to enabling the development of real-world AI applications while ensuring high standards of quality and privacy. Engineers will work on a platform that supports generative AI applications and enhances the ML lifecycle from experimentation to deployment.

• 10+ years of work experience in machine learning infrastructure, distributed data systems, or a related field.
• 10+ years of experience building and shipping large-scale data or ML infrastructure and platforms in production.
• Extensive experience architecting and delivering large-scale distributed data or ML infrastructure that multiple teams or products depend on in production.
• Deep systems engineering skills with strong Python and a systems language, preferably Rust.
• Deep familiarity with columnar and lakehouse formats such as Parquet, Iceberg, Delta, or Lance.
• Strong working knowledge of the end-to-end ML workflow and how training and inference consume data.
• Familiarity with modern ML and generative techniques such as transformers and fine-tuning.
• B.S., M.S., or Ph.D. in Computer Science, Computer Engineering, or equivalent practical experience.

• Set the technical direction for the ML infrastructure platform.
• Own and solve the hardest system-level problems related to data ingestion, versioning, and governance.
• Design highly available and user-friendly systems while mentoring other engineers.
• Collaborate with multiple teams to align on technical direction and deliver solutions.
• Architect data systems that effectively serve the needs of training and inference.
• Contribute to the development of data-loading frameworks for machine learning.

• Employees at Apple are often offered comprehensive benefits that support physical and mental well-being—flexible medical plans, confidential counseling, onsite wellness centers at major campuses, and resources for fitness and daily life. Families typically receive fertility support, paid parental leave with gradual return, caregiving leave, and dependent-care guidance, while financial perks commonly include stock grants (with purchase discounts), 401(k) matching, and income-protection coverage. Employees also see robust time off, Apple University learning and tuition reimbursement, donation matching and paid volunteer hours, and deep product and partner discounts.