Principal On-Device AI Engineer – Real-Time Game Inference

Unity Technologies; Mountain View (CA); On-site; USD 278,100 - 417,100; Full time

Benefits offered by this job

Comprehensive health, life, and disability insurance

Employee stock ownership

Competitive retirement/pension plans

Generous vacation and personal days

Job summary

Unity Technologies is looking for a Principal Machine Learning Engineer in Mountain View, CA, to lead the integration of AI-driven models into game engines. The role involves optimizing multi-modal models for on-device use, ensuring performance within battery and memory constraints.

The ideal candidate will have extensive software/ML engineering experience with a strong focus on deploying AI models and leading engineering teams. Unity offers comprehensive benefits and is an equal opportunity employer committed to diversity.

Qualifications

• 8+ years in software/ML engineering, with at least 4 years focused on on-device/edge inference.
• Proven production deployment of transformer and/or diffusion-based models.
• Hands-on experience deploying models through WebGPU.

Responsibilities

• Own the end-to-end optimization pipeline for models.
• Evaluate and drive adoption of WebGPU-targeted inference runtimes.
• Lead and mentor a team of engineers.

Skills

Software/ML engineering experience

On-device inference

Low-level performance engineering

GPU/compute API proficiency

Model optimization techniques

Technical leadership

Tools

WebGPU

TensorFlow.js

ONNX Runtime

Job description

Unity Technologies is looking for a Principal Machine Learning Engineer in Mountain View, CA, to lead the integration of AI-driven models into game engines. The role involves optimizing multi-modal models for on-device use, ensuring performance within battery and memory constraints.

The ideal candidate will have extensive software/ML engineering experience with a strong focus on deploying AI models and leading engineering teams. Unity offers comprehensive benefits and is an equal opportunity employer committed to diversity.