The On-Device Machine Learning team at Apple is seeking an ML Infrastructure Engineer focused on ML user experience APIs and integration. This role involves developing new ML model conversion and authoring APIs that serve as the main entry point into Apple's ML infrastructure. The engineer will work on optimizing and integrating ML tools into model repositories, enabling efficient model ingestion and implementation. This position offers the opportunity to influence how ML developers interact with Apple's end-to-end inference stack.

• Bachelor's degree in Computer Sciences, Engineering, or a related subject area.
• Highly proficient in Python programming, with familiarity in C++.
• Proficiency in at least one ML authoring framework, such as PyTorch, MLX, or JAX.
• Strong understanding of ML fundamentals, including common architectures such as Transformers.
• Hands-on experience with ML inference optimizations, such as quantization and pruning.
• Strong communication skills to connect with multi-functional audiences.
• Experience with C++, Swift, and/or GPU programming paradigms.
• Familiarity with QAT and other compression and quantization techniques using PyTorch workflows.
• Experience designing Python APIs and deploying production-grade Python packages.
• Experience with MLIR/LLVM or similar compiler toolchains.
• Familiarity with Hugging Face or other model repositories.

• Develop and use APIs that enable ML engineers to efficiently author and convert ML models for Apple platforms.
• Integrate Apple's ML tools and APIs into internal and external model repositories.
• Ideate, design, and stress test optimizations required to support ML models.
• Showcase the practical power of Apple's authoring and runtime APIs by creating capable models.
• Shape how ML developers experience Apple's end-to-end inference stack from model creation to deployment.

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