We are seeking a Machine Learning Engineer to join our ML Compute team at Apple, focusing on enhancing the efficiency, scalability, and reliability of model training and inference workloads in the cloud. The role involves leading the integration of large-scale ML workloads with cloud infrastructure while collaborating with ML engineers, infrastructure engineers, and researchers. The team is dedicated to supporting the training of foundation models that enhance Apple products through advanced capabilities in understanding and generating various modalities. Ideal candidates will be passionate about building systems that advance deep learning and improve user experiences.

• 5+ years of experience in software engineering, ML infrastructure, or related domains.
• Hands-on experience with machine learning workflows, including training, evaluation, and inference at scale.
• Proficiency in Python and experience with at least one major ML framework such as PyTorch or JAX.
• Experience with cloud-based infrastructure and distributed systems, including containers, orchestration, storage, and networking.
• Bachelor's degree in Computer Science, Engineering, or a related field.
• Experience working with accelerator-based systems, including performance tuning and debugging of ML workloads.
• Hands-on experience with distributed training or inference at scale.
• Experience optimizing large-scale ML systems, including bottleneck analysis across compute, memory, and networking.
• Familiarity with profiling, tracing, and benchmarking tools for ML workloads.
• Experience building or operating ML infrastructure using containerization and orchestration frameworks.

• Lead the integration of large-scale ML workloads with cloud infrastructure.
• Collaborate cross-functionally with ML engineers, infrastructure engineers, and researchers.
• Optimize performance and improve system efficiency for model training and inference workloads.
• Drive high utilization of accelerator resources in the cloud.
• Support the training of foundation models with advanced capabilities for Apple products.

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