Search jobs > Seattle, WA > Machine learning engineer

AIML - Machine Learning Engineer, Core ML Model Conversion (ODML) Machine Learning Platform & Infrastructure

Apple
Seattle
Full-time

Summary :

Are you passionate about revolutionizing on-device AI and shaping the future of ML deployment? Join the team behind Core ML, the technology that drives transformative ML features across Apple's ecosystem and beyond.

Our cutting-edge technology leverages the full potential of Apple Silicon to execute neural networks with unprecedented speed and efficiency on Macs, iPhones, Apple Watch, and Vision Pro.

Our team played a pivotal role in the groundbreaking Apple Intelligence features recently unveiled at WWDC, showcasing the cutting-edge capabilities of on-device AI.

We were instrumental in getting the 3B on-device model running on device efficiently. We enable intelligent features in core Apple applications like Camera, Siri, Keyboard, FaceTime, Spatial Computing etc and empower professional-grade third-party applications like Adobe Photoshop, Pixelmator etc to leverage the power of on-device AI.

By harnessing the full potential of Apple's custom-designed CPU, GPU, and Neural Engine, we make it possible for these applications to deliver exceptional AI-powered experiences with unparalleled performance and efficiency.

As a key player in our team, you'll be at the forefront of bridging the gap between the training of state-of-the-art deep learning models and their efficient execution on Apple devices.

Your expertise will be crucial in converting complex computational graphs from frameworks like PyTorch and JAX into optimized, Apple-friendly format that unleashes the full potential of our hardware.

You will work on bringing state of the art models, of varying sizes to run on device with high performance (low latency, memory and power).

We're seeking a highly motivated individual with a deep understanding of ML models and a passion for bringing cutting-edge models from ML research and different domains (vision, image / text generation, audio etc) into real-world applications on device.

If you have a track record of optimizing and deploying models, writing high-quality code, and delivering impactful libraries to a broad user base, we want to hear from you.

In this highly visible role, you'll collaborate with innovative teams across Apple and industry-leading external partners like Adobe, Hugging Face, and Meta.

You will work with them to export and efficiently deploy deep learning models on Apple devices. Your contributions will be instrumental in shaping the future of on-device AI, and you'll play a key role in Apple's open-source initiatives, directly impacting the global AI developer community.

Key Qualifications : Description :

Description :

We are the team that develops Core ML Tools, an open source python library for converting PyTorch and TensorFlow models to Core ML and optimizing models for performance.

If you enjoy playing with the building blocks and architecture of machine learning models, and are strong at understanding the mathematical operations making these models and manipulating the computational graph to optimize for speed / execution, then you are going to have fun in this role! Responsibilities include :

  • Performing model conversion from PyTorch, among other libraries, to the Core ML model format
  • Running and benchmarking models. Understanding the effect of computational graph representation on the model execution performance on Neural Engine, GPU, CPU.
  • Proficient in setting up and running open source ML models (e.g. Hugging Face), understanding ML pipelines and reasoning on which parts should be part of the model, and which ones should be outside of the model as pre-processing and post processing steps
  • Adding graph passes for improving performance. Publishing examples of models that are converted in "performant" ways (example : the Apple Stable diffusion open source library)* Collaborate effectively with developers (internal and external).

Be an active member of the open source CoreMLTools community on Github, interacting with developers, addressing GitHub issues etc* Implementing new operations / layers for neural networks

Improving model optimization documentation, writing examples, tutorials and guides Join us in shaping the future of on-device AI and be part of the team that's redefining what's possible with machine learning on Apple devices!

Additional Requirements :

30+ days ago
Related jobs
Promoted
TikTok
Seattle, Washington

We are looking for ML engineers who are excited to grow their business understanding, build highly scalable and reliable software/infrastructure, partner across functions with global teams, and make big impacts. Work closely with cross-functional teams to improve the platforms that support our AI mo...

T-Mobile
Bellevue, Washington

As a Machine Learning Engineering Intern, you are required to understand enabling systems for coding, deploying, and maintaining large-scale machine learning models throughout their lifecycle. You should understand the entire machine learning (ML) lifecycle, including software engineering, data coll...

Promoted
TikTok
Seattle, Washington

Proficiency in modern machine learning theories and applications, including ensemble trees, deep neural networks, transfer/multi-task learning, reinforcement learning, graph theory, and unsupervised learning. We achieve our mission by a) developing state-of-art Machine Learning (ML) solutions to pre...

Rokt
Seattle, Washington

You will be working with our engineering and product teams to design, build and productionize proprietary machine learning models to solve different business challenges including smart bidding, lookalike modeling, forecasting, and etc. Collaborate closely with product managers and other engineers to...

Promoted
TikTok
Seattle, Washington

Build and design large-scale machine learning algorithms to model the real-time interests of creators to improve conversion efficiency. We create to inspire - for you, for us, and for more than 1 billion users on our platform. The e-commerce alliance team aims to serve merchants and creators in the ...

Apex Systems
Redmond, Washington

Apex is hiring a Machine Learning Software Developer to work with Researchers to take their ML models and optimize them. Candidates with strong C++, Python, and firmware experience who want experience with cutting-edge machine learning models, and electromyography with wearable device development ar...

TikTok
Seattle, Washington

Participate in the development and iteration of Ads algorithms by using Machine Learning, including ads query understanding, ads targeting, ads ranking, model serving reliability, etc. Good theoretical grounding in the machine and deep learning concepts and techniques (CNN/RNN/LSTM,. Familiar with t...

FIS
Bellevue, Washington

Integrate ML models with the SaaS platform and other services and tools, such as the model registry, feature store, data lake, and event streams. As a Machine Learning Data Engineer, you will tackle challenges ranging. We are looking for talented Machine Learning Engineer. The ideal candidate will h...

TikTok
Seattle, Washington

Minimum Qualification - Bachelor degree in computer science or a related major, with 5+ years of related work experience;- Strong software development experience with C++, Python or other programming language;- Experience in one or more of the following areas: applied machine learning, machine learn...

VirtualVocations
Seattle, Washington

Applied Machine Learning Researcher specializing in Generative Models. Master's degree in Electrical Engineering, Computer Science, or a related disciplineProven experience (2+ years) in applied research focused on generative models for audio or musicProficiency in Python and deep learning framework...