The On-Device Machine Learning team at Apple is seeking an experienced ML Infrastructure Engineer to build an execution engine and compilation toolchain for optimizing machine learning models on Apple products. This senior role involves integrating compiler technology with runtime components and hardware to enhance ML execution across various devices. The ideal candidate will be passionate about machine learning and system software engineering, contributing to the development of innovative AI applications. This position offers the opportunity to impact the future of intelligent experiences on Apple platforms.

• Bachelor's degree in Computer Science, Engineering, or a related subject area.
• 5+ years of hands-on experience in software engineering.
• Highly proficient in C++ with familiarity in Python and Swift.
• Understanding of operating systems and embedded programming.
• Sound understanding of machine learning fundamentals, including architectures like Transformers.
• Good communication skills for interacting with multi-functional teams.
• Experience with on-device ML stacks such as TFLite, ONNX, or ExecuTorch.
• Experience with open source machine learning models like Mistral, Phi, or Huggingface.
• Experience with compiler stacks such as MLIR, LLVM, or TVM.
• Experience with ML authoring frameworks like PyTorch, TensorFlow, or JAX.
• Experience with machine learning accelerators and GPU programming.

• Build and optimize the execution engine and compilation toolchain for machine learning models.
• Integrate compiler technology with runtime components and hardware libraries.
• Make critical decisions that affect project direction and outcomes.
• Collaborate with cross-functional teams to enhance the developer experience for machine learning.
• Contribute to the development of benchmarking and debugging toolchains for ML workflows.

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