As a Senior/Staff Engineer on the Foundation Model Compute Infrastructure team, you will be responsible for designing and building large-scale infrastructure that supports foundation model training, fine-tuning, evaluation, and inference. This role involves developing model inference and fine-tuning services while collaborating closely with foundation model researchers and engineers. You will focus on improving reliability, performance, scalability, and developer productivity across Apple's AI workloads. Your contributions will help shape the future of AI at Apple.

• 5+ years of industry experience building large-scale distributed systems or cloud infrastructure
• Experience with distributed ML training or inference systems
• Strong programming skills in Python, Go, C++, or similar systems languages
• Experience with accelerator infrastructure such as TPU or GPU
• Experience with Kubernetes, container orchestration, or large-scale cluster management systems
• Bachelor's degree in Computer Science, Engineering, or related field
• Experience building schedulers, resource managers, or orchestration systems for distributed workloads
• Familiarity with frameworks such as JAX, PyTorch, TensorFlow, Ray, Pathways, or vLLM
• Experience operating large-scale multi-tenant infrastructure in cloud or hybrid environments
• Background in performance optimization, fault tolerance, or resource efficiency for large distributed systems
• Strong expertise in distributed systems, scalability, reliability, and performance engineering
• Experience designing backend services or infrastructure platforms operating at production scale
• MS or PhD in Computer Science, Engineering, or related field

• Design and build large-scale infrastructure for foundation model training and inference.
• Develop model inference and fine-tuning services.
• Onboard and benchmark new accelerators.
• Collaborate with foundation model researchers and engineers to enhance system performance.
• Improve reliability, scalability, and developer productivity across AI workloads.

• 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.