Impress employers and recruiters.
Choose from hundreds of resume examples.

Target Resume Now

About the Role

We are building the next generation of AI-driven game experiences, running generative models on-device. As a Senior Machine Learning Engineer for On-Device & Mobile AI, you will optimize state-of-the-art multi-modal models to run fast, small, and reliably on mobile and constrained hardware. This hands-on role involves owning the optimization and deployment of the inference stack, directly shaping the player experience.

Responsibilities

Own the optimization pipeline for models: model export, graph transformation, operator fusion, memory-layout planning, and hardware-specific tuning.
Apply quantization, weight sharing, pruning, and knowledge distillation to meet latency, memory, and power budgets.
Perform low-level performance work, including writing and tuning WebGPU compute shaders and native kernels.
Profile performance using browser and platform tools to eliminate bottlenecks.
Apply efficiency techniques like dynamic resolution and reduced-step diffusion samplers.
Work with WebGPU-targeted inference runtimes and extend or build glue code.
Build parts of the integration between the ML runtime and the game engine.
Build supporting engineering for components, including model packaging and asset pipelines.
Partner with research scientists to turn novel architectures into deployable, debuggable, and fast on-device implementations.
Provide feedback to research on hardware constraints and op-support gaps.
Track and assess breakthroughs in efficient inference.
Contribute to engineering best practices, code-review standards, and performance-regression gates.
Support a culture of measurement by tracking KPIs for latency, quality, memory, and power.
Partner with platform engineers, product managers, and runtime teams.
Share knowledge and mentor junior and mid-level engineers.

Requirements

5+ years in software/ML engineering, with significant focus on on-device/edge inference or real-time, performance-critical systems.
Production deployment of transformer- and/or diffusion-based models on mobile, desktop, or embedded hardware.
Hands-on experience with at least one major inference runtime (ONNX Runtime / ORT Web, CoreML, TFLite, ExecuTorch).
Working understanding of operator fusion, memory layout, and runtime scheduling.
Solid command of at least one GPU/compute API (WebGPU/WGSL, Metal, Vulkan, D3D12, or CUDA) and associated profiling tools.
Ability to read frame captures and kernel traces to identify performance bottlenecks.
Working knowledge of model-optimization techniques (quantization, weight sharing, pruning, distillation).
Understanding of target hardware (mobile SoCs and/or desktop/laptop GPUs).
Strong Python for export pipelines and training-side tooling.
Working fluency with models to read architectures, modify for deployment, and reason about accuracy trade-offs.
Collaborative working style with clear communication and reliable delivery.

Skills

WebGPU
WGSL
Metal
Vulkan
D3D12
CUDA
ONNX Runtime
ORT Web
CoreML
TFLite
ExecuTorch
Python
TypeScript
JavaScript
Quantization
Weight sharing
Pruning
Knowledge distillation
Operator fusion
Memory layout planning
Hardware-specific tuning
Profiling
Frame capture analysis
Kernel trace analysis
Transformer models
Diffusion models
Vision-language models (VLMs)

Location

Remote

Work Type

Full-time

Experience Level

Senior

Salary/Compensations

$167,200.00 - $250,800.00

Benefits

Comprehensive health, life, and disability insurance
Commute subsidy
Employee stock ownership
Competitive retirement/pension plans
Generous vacation and personal days
Support for new parents through leave and family-care programs
Office food snacks
Mental Health and Wellbeing programs and support
Employee Resource Groups
Global Employee Assistance Program
Training and development programs
Volunteering and donation matching program

About the Company

Unity [NYSE: U] is the world’s leading game engine, powering play for more than 3 billion consumers each month.
The top mobile games in the world, the most played PC indie titles, the most innovative console games, and virtually all of the top XR and Web Games are developed, deployed, and grown in Unity.
Unity also enables teams across industries like automotive, manufacturing, and healthcare to design, simulate, and collaborate in 3D — closing the gap between ideas and reality.
We are committed to creating a workplace that fosters collaboration and teamwork that allows you to harness your unique skills.
Join us in creating industry tools to help creators of all levels bring their projects to reality.

Equal Opportunity

Unity is a proud equal opportunity employer.
We are committed to fostering an inclusive, innovative environment and celebrate our employees across age, race, color, ancestry, national origin, religion, disability, sex, gender identity or expression, sexual orientation, or any other protected status in accordance with applicable law.
Our differences are strengths that enable us to support the growing and evolving needs of our customers, partners, and collaborators.
If you have a disability that means there are preparations or accommodations we can make to help ensure you have a comfortable and positive interview experience, please fill out this form to let us know.
Pursuant to the San Francisco Fair Chance Ordinance, we will consider for employment qualified applicants with arrest and conviction records.