Jobs in the United States

25+ jobs

Save this search

Added 20 hours ago
Silicon Physical Design Engineer V
Actalent
Menlo Park, CA
We are seeking a Silicon Physical Design Engineer V to join our team and perform power analysis and optimizations in ASIC for AR/VR products. The ideal candidate will have expertise in…
Added 20 hours ago
Software Engineer II - AI/ML, Neuron Inference, Neuron Inference
Amazon
Cupertino, CA
…technology layers - from frameworks and kernels through to compiler, runtime, and collectives. We not only optimize performance on open weight models but also contribute to future…
Added 21 hours ago
Applied Scientist, Neuron ARG
Amazon
Seattle, WA
…to raise the code quality bar in our state-of-the-art deep learning compiler stack. This stack is designed to optimize application models across diverse domains, including Large Language…
Added 1 day ago
Software Engineer II - AI/ML, Neuron Inference, Neuron Inference
Annapurna Labs (U.S.) Inc.
Cupertino, CA
…technology layers - from frameworks and kernels through to compiler, runtime, and collectives. We not only optimize performance on open weight models but also contribute to future…
Added 2 days ago
Technical Content Writer
Efficient Computer
TX · Austin, TX · San Francisco, CA
Embedded machine learning, digital signal processing, or real-time control
Added 2 days ago
Compiler Software Engineering Intern - Summer 2027
NVIDIA USA
WA
…Computing Engineering grad student to support our NVIDIA Back-End Compiler Team for a 2021 internship! Our team is responsible for constructing and emitting the highest performance GPU…
Added 3 days ago
Sr. Software Engineer- AI/ML, AWS Neuron Distributed Training - Performance Optimization
Amazon
Seattle, WA
learning accelerators and the Trn3/Trn2/Trn1 and Inf2/Inf1 servers that use them. This role is for a software engineer in the Distributed Training team for AWS Neuron. This role is…
Added 4 days ago
Applied Scientist, Neuron ARG
Annapurna Labs (U.S.) Inc.
Seattle, WA
…to raise the code quality bar in our state-of-the-art deep learning compiler stack. This stack is designed to optimize application models across diverse domains, including Large Language…
Added 4 days ago
Senior Software Engineer, ML API – AI Inference
Cerebras Systems
Sunnyvale, CA
This role sits at the intersection of machine learning systems, API design, model serving, and distributed systems. You will translate evolving model capabilities and customer requirements…
Added 4 days ago
Compiler Engineer - New Grad
SambaNova
Austin, TX · San Jose, CA
As a new grad Compiler Engineer, you'll work on core areas of our compiler, including the IR and pass infrastructure, model lowering and partitioning, tensor tiling, memory management, and…
Added 5 days ago
On-Device ML Compiler Engineer, Model Compilation, Graphics, Games and Machine Learning
Apple
Cupertino, CA
We’re seeking a highly motivated software engineer who is creative, versatile, and passionate about machine learning, common compiler optimizations, and system software engineering in the…
Added 5 days ago
Senior Software Engineer, Performance, SOC Architect
Google
Sunnyvale, CA
As a part of this team, you will be focused around driving continuous improvements to the machine learning software/hardware stacks through providing insightful performance debugging for…
Added 5 days ago
TI
Mid-Level Machine Learning Engineer
TETRAMEM INC
San Jose, CA
- $110,000 to $200,000 USD / year
Develop, optimize, and deploy lightweight machine learning models for edge AI applications, particularly for audio processing.
Added 5 days ago
TI
Senior Machine Learning Engineer
TETRAMEM INC
San Jose, CA
- $200,000 to $280,000 USD / year
Develop, optimize, and deploy lightweight machine learning models for edge AI applications, particularly for audio processing.
Added 5 days ago
TI
Compiler Engineer
TETRAMEM INC
San Jose, CA
- $160,000 to $300,000 USD / year
Design, develop, and maintain compiler toolchains that translate machine learning models from industry-standard frameworks into optimized workloads for TetraMem’s analog in-memory computing…
Added 5 days ago
TI
US 2026 Software - Compiler Engineer Intern
TETRAMEM INC
San Jose, CA
- $35 to $45 USD / hour
We are seeking a Compiler Engineer intern to support the Compiler Software team in developing the compiler and runtime for in-memory machine learning accelerators at TetraMem.
Added 5 days ago
Software Engineer I – Compiler & Runtime
TETRAMEM INC
San Jose, CA
Collaborate with compiler, machine learning, hardware, and validation teams to resolve software issues and improve overall system performance.
Added 5 days ago
TI
Field Application Engineer
TETRAMEM INC
San Jose, CA
- $120,000 to $180,000 USD / year
Collaborate with Software, Compiler, Hardware, Machine Learning, and Product Engineering teams to investigate customer issues and drive timely resolutions.
Added 5 days ago
TI
System Engineer
TETRAMEM INC
San Jose, CA
- $110,000 to $250,000 USD / year
Develop, optimize, and deploy lightweight machine learning models for edge AI applications, particularly for audio processing.
Added 5 days ago
ReRAM Memory and AI Development Engineer
onsemi
Gresham, OR
The role combines hands-on ReRAM technology development with the broader application of machine learning, generative AI, physics-informed AI, and AI agents across semiconductor engineering.…
Added 5 days ago
BV
Parallel Computing Engineer
Bright Vision Technologies
Remote, United States
- $130,000 to $180,000 USD / year
Bright Vision Technologies is seeking a highly experienced Parallel Computing Engineer with 10+ years of experience in High-Performance Computing (HPC), GPU programming, and parallel…
Added 7 days ago
AI Systems Engineer, Hardware Architecture
Meta
Washington, DC
…Please note that Meta may leverage artificial intelligence and machine learning technologies in connection with applications for…
Added 7 days ago
AI Systems Engineer, Hardware Architecture
Meta
Phoenix, AZ
…Please note that Meta may leverage artificial intelligence and machine learning technologies in connection with applications for…
Added 7 days ago
AI Systems Engineer, Hardware Architecture
Meta
Hartford, CT
…Please note that Meta may leverage artificial intelligence and machine learning technologies in connection with applications for…
Added 7 days ago
AI Systems Engineer, Hardware Architecture
Meta
Charleston, WV
…Please note that Meta may leverage artificial intelligence and machine learning technologies in connection with applications for…

Apple

On-Device ML Compiler Engineer, Model Compilation, Graphics, Games and Machine Learning

Cupertino, CA, United States

Apply About employer

Applications open in a new tab.

Job details

Posting language: English
Posted: Sep 3, 2026

Original job description

Role Number: 200658150-0836

Summary

Imagine being at the forefront of an evolution where innovative AI meets the elegance of Apple silicon. The On-Device Machine Learning team transforms groundbreaking research into practical applications, enabling billions of Apple devices to run powerful AI models locally, privately, and efficiently. We stand at the unique intersection of research, software engineering, hardware engineering, and product development, making Apple the leading destination for machine learning innovation.

Our team builds the essential infrastructure that enables machine learning at scale on Apple devices. This involves onboarding innovative architectures to embedded systems, developing optimization toolkits for model compression and acceleration, building ML compilers and runtimes for efficient execution, and creating comprehensive benchmarking and debugging toolchains. This infrastructure forms the backbone of Apple’s machine learning workflows across Camera, Siri, Health, Vision, and other core experiences, contributing to the overall Apple Intelligence ecosystem. If you are passionate about the technical challenges of running sophisticated ML models across all devices, from resource-constrained devices to powerful clusters, and eager to directly impact how machine learning operates across the Apple ecosystem, this role presents a great opportunity to work on the next generation of intelligent experiences on Apple platforms.

Our group is seeking an ML Infrastructure Engineer, with a focus on model compilation. The role entails working closely with model authoring, runtime, and performance teams to ensure that models can bring to bear the full capabilities of the hardware.

Description

We’re building an end-to-end developer experience for machine learning development that employs Apple’s vertical integration. This allows developers to iterate on model authoring, optimization, transformation, execution, debugging, profiling, and analysis. This role focuses on the core runtime for execution across a wide variety of devices and use cases.

We’re seeking a highly motivated software engineer who is creative, versatile, and passionate about machine learning, common compiler optimizations, and system software engineering in the fast-paced and dynamic field of machine learning. We have an MLIR-based compiler stack, and use it to target the neural engine, GPU, and CPU in order to harness the full capabilities of the system for ML workflows and execution.

Minimum Qualifications

3-5 years working on MLIR-based compilers.
Familiarity with common ML model architectures, execution schemes, and operations.
Familiarity with C++ Familiarity with PyTorch or related training frameworks

Preferred Qualifications

Familiarity with Swift.
Familiarity with programming paradigms for the GPU, CPU, and Neural Engine.
Familiarity with writing kernels for ML model execution.

Apple is an equal opportunity employer that is committed to inclusion and diversity. We seek to promote equal opportunity for all applicants without regard to race, color, religion, sex, sexual orientation, gender identity, national origin, disability, Veteran status, or other legally protected characteristics.

Apply About employer

More about this job

Skills

ML
Machine learning
Debugging
C++
PyTorch
Swift

Report this job