Principal Machine Learning Engineer

Unity; Mountain View (CA); On-site; USD 190,000 - 270,000; Full time

Job summary

Unity is seeking a Principal Engineer for On-Device AI Inference & Systems to advance multi-modal models (transformers and diffusion networks) running entirely within our game engine runtime.

You will own the inference and integration stack end-to-end—from exporting and optimizing trained checkpoints to kernel-level tuning and shipping features that run at interactive frame rates within strict memory and power budgets.

Job description

We are building the next generation of AI-driven game experiences, running generative models on-device, right where the players are -- on phones, tablets, laptops, and desktops. Our games run inside a modern, browser-native runtime (built on technologies such as WebGPU and WebNN), so the models that power these experiences must be deployed and accelerated entirely within that runtime. As our Principal Engineer for On-Device AI Inference & Systems, you will be the foremost engineering authority on taking state-of-the-art multi-modal models (transformers and diffusion networks) and making them run fast, small, and reliably within that runtime, fully integrated into a production game engine.

This is a deeply hands-on, high-impact engineering role. You will own the inference and integration stack end-to-end - from the moment a trained checkpoint leaves research, through export, optimization, and kernel-level tuning, to a shipped feature running inside the engine at interactive frame rates within a fixed memory and power budget. You will set the engineering standards, drive the architecture of the runtime and integration layers,