Staff Machine Learning Engineer

Unity is building next-generation AI-driven gaming experiences with generative world models, neural rendering, and multi-modal understanding — this hands-on technical leadership role brings cutting-edge computer vision and multi-modal models into production for billions of players.

$218,400 – $327,600 / yr

Location

Mountain View, CA

On-site

Seniority

Staff

Compared with 41 other Engineering roles: Common across similar roles Unique to this role

What you'll do

Responsibilities

• Establish technical vision and roadmap for computer vision and multi-modal AI (transformers, diffusion models, vision-language models)
• Drive design of systems for image/video understanding, generation, segmentation, detection, and multi-modal reasoning
• Make architectural decisions balancing quality, latency, and cost across cloud, server, and on-device targets
• Own the full pipeline from research prototypes to production: training, fine-tuning, distillation, serving
• Apply compression, quantization, pruning, and knowledge distillation for constrained environments Unique
• Lead and mentor ML engineers; define and track KPIs for model quality, accuracy, latency, memory, and cost

What you'll need

Experience

• 6+ years ML engineering with significant depth in computer vision and/or multi-modal modeling
• Production experience with transformer-based and diffusion vision models (ViT, CLIP, Stable Diffusion, DETR, SAM)
• Strong full model lifecycle knowledge: data curation, training, evaluation, serving at scale
• Familiarity with efficient attention, diffusion samplers, multi-modal fusion, and vision-language alignment Unique
• Strong Python and PyTorch skills; proven technical leadership track record
• Preferred: world-model/video-generation/neural rendering pipelines (NeRF, 3DGS), on-device deployment (CoreML, TFLite, ONNX), real-time graphics or game engine background Common

#J-18808-Ljbffr