Staff Machine Learning Engineer

Posted 2 days ago

workman labsMountain View (CA)

SENIORITY

Manager

SALARY

$218,400 per year

Apply

About the role

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

Before you apply

Applying takes about a minute. These four things decide how fast it moves after that.

Your profile is current

It's what we read first. Occupations, seniority and locations matter more than a long history.

Two examples you can talk through

Not a portfolio — just two pieces of work where you can explain the decisions and what you'd change.

A number in mind

What you're on now and what would make you move. We negotiate better when we know both.

Your notice period

Employers plan around it, and it's the question that stalls offers most often.

Once you apply, someone reads it and calls you before anything reaches the employer — usually within two working days.

More like this