Unknown Company

Staff Machine Learning Engineer

mountain view, ca • Posted 4 days ago
Onsite Full Time Electrical & Energy Engineering

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
Back to Job Search