Senior On-Device ML Engineer for Real-Time Multimodal AI

LE130 Unity Technologies SF; United States; Remote; USD 218,000 - 284,000; Full time

Benefits offered by this job

Health and life insurance

Commuter subsidy

Employee stock ownership

Retirement plans

Generous vacation and personal days

Parental leave

Office snacks

Mental health programs

Employee Resource Groups

Training and development programs

Volunteer and donation matching

Job summary

Unity Technologies is seeking a Senior Machine Learning Engineer for On-Device & Mobile AI to optimize and deploy multi-modal models on mobile and constrained hardware. You will push federated optimization, export pipelines, and low-level GPU kernels to meet strict latency and memory budgets, collaborating with runtime and game engine teams.

The role emphasizes hands-on development, profiling, and cross-platform integration, with a focus on transformer- and diffusion-based models deployed

Qualifications

• 5+ years in software/ML engineering.
• Production deployment of transformer- and/or diffusion-based models on mobile/desktop hardware.
• Hands-on experience with at least one major inference runtime (ONNX Runtime / ORT Web, CoreML, TFLite, ExecuTorch).
• Low-level performance engineering with GPU/compute APIs and profiling tools.
• Familiarity with model-optimization: quantization, pruning, distillation.

Responsibilities

• Own the optimization pipeline for the models you ship: export, graph transformation, operator fusion, memory-layout planning, hardware tuning.
• Write/tune WGSL compute shaders and native kernels; profile with browser/platform tools; eliminate bottlenecks.
• Collaborate across runtimes and game engine integration: scheduling, memory pooling, zero-copy buffers, frame-budget management.
• Partner with research to productionize CV and multi-modal architectures for on-device use.
• Contribute to engineering standards, benchmarks, and CI to ensure performance.

Skills

Python

WebGPU/WGSL

C++/Objective-C/Swift

Profiling

On-device inference

Tools

ONNX Runtime Web

CoreML

TFLite

ExecuTorch

Job description

Unity Technologies is seeking a Senior Machine Learning Engineer for On-Device & Mobile AI to optimize and deploy multi-modal models on mobile and constrained hardware. You will push federated optimization, export pipelines, and low-level GPU kernels to meet strict latency and memory budgets, collaborating with runtime and game engine teams.

The role emphasizes hands-on development, profiling, and cross-platform integration, with a focus on transformer- and diffusion-based models deployed