The ML Integration Engineer will work within the On-Device Machine Learning team at Apple, focusing on integrating machine learning workflows to enhance user experience and performance. This role involves collaborating with various teams to ensure seamless functionality across Apple's ML deployment stack. The engineer will also be responsible for optimizing ML models and developing tools for analysis and debugging. This position offers the opportunity to influence how machine learning operates across Apple devices.

• Bachelor's degree in Computer Sciences, Engineering, or a related discipline.
• Proficient in Python programming.
• Some familiarity with C++ is required.
• Proficiency in at least one ML authoring framework, such as PyTorch, MLX, or JAX.
• Understanding of ML fundamentals, including common architectures such as Transformers.
• Understanding of GPU programming paradigms.
• Strong communication skills, including the ability to communicate with cross-functional audiences.
• Experience with C++, Swift.
• Experience with GPU kernel optimizations.
• Experience with MLIR/LLVM or similar compiler toolchains.
• Familiarity with Hugging Face or other model repositories.

• Ensure Apple's inference stack allows for the integration of ML workflows end-to-end.
• Partner with teams across the ML deployment stack to enhance user experience and performance.
• Work on model-side updates, ML frameworks export, and custom kernels.
• Optimize compilers and develop analysis and debugging tools.
• Spearhead the integration of the latest ML models across hardware targets.
• Shape how ML developers experience Apple's end-to-end inference stack.

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