On-Device ML Integration Engineer, Graphics, Games and Machine Learning
At a glance
AI generatedTL;DR
As an On-Device Machine Learning Integration Engineer at Apple, you will join a dynamic team dedicated to transforming advanced AI research into practical applications that run efficiently on Apple devices. Your primary responsibilities include ensuring seamless integration of ML models across the inference stack for optimal performance and user experience. You will collaborate closely with various teams, from model developers to runtime engineers, to optimize ML workflows through model-level transformations, custom operations, and compiler optimizations. Utilizing Python and proficiency in frameworks like PyTorch or JAX, you will integrate cutting-edge ML tools into internal and external repositories, demonstrating peak efficiency and performance across diverse hardware targets. This role requires a deep understanding of GPU programming paradigms and experience with MLIR/LLVM for compiler toolchains, making it ideal for those passionate about shaping the future of machine learning on Apple platforms.
Skills
What you'll do
- Ensure functional and performant integration of Apple’s ML models across the inference stack.
- Develop optimizations including model-level transformations, custom operations, and compiler optimizations to enhance inference efficiency.
- Integrate cutting-edge ML models into Apple’s infrastructure with peak performance.
- Partner with teams to stress-test model ingestion in internal and external repositories for maximum efficiency.
- Spearhead the validation of Apple’s inference stack using integrated models.
What we're looking for
- Bachelor’s degree in Computer Science, Engineering, or related field.
- Proficiency in Python and familiarity with C++.
- Expertise in at least one ML authoring framework like PyTorch, MLX, JAX.
- Knowledge of GPU programming paradigms and ML fundamentals.
- Experience developing optimizations across the ML pipeline for efficiency.
Employer
About Apple Inc
Apple Inc. is a multinational technology company known for designing and manufacturing consumer electronics, software, and online services, including the iPhone, Mac, iPad, and App Store. Industry: Consumer Electronics & Software
Apple Inc currently has 255 open roles on FindRole.
Listed pay typically runs $171,600–$272,100 across 182 roles with salary data.
Most-posted roles
- Software Development Engineer 10
- Apple Business Systems Engineer Manager 8
- iPad Touch Electrical Engineer 3
- Machine Learning Engineer, Apple Store Online 3
- Manager, Machine Learning, Apple Store Online 3