Unity
Staff Machine Learning Engineer
$172K–$284K
Mountain View, CA, USA · Staff · Full-time · Posted Sep 17, 2026
- Experience
- 6+ years in ML engineering.
About the job
Leads the development and production deployment of computer-vision and multimodal models for AI-driven interactive game experiences across cloud, server, and on-device platforms.
What you'll do
- Set technical direction and design models for image/video understanding and generation, segmentation, detection, dense prediction, and multimodal reasoning.
- Bring research models into production, overseeing training, fine-tuning, evaluation, optimization, export, and serving.
- Lead and mentor ML engineers and partner with research, platform, product, and runtime teams.
What they ask for
- Significant depth in computer vision and/or multimodal modeling.
- Production experience with transformer- and diffusion-based vision models.
- Strong command of the model lifecycle, including data curation, training, evaluation, and serving at scale.
- Strong Python, PyTorch, software engineering fundamentals, and a track record of technical leadership.
Nice to have
- Experience with world-model, video-generation, or neural-rendering pipelines.
- Experience deploying optimized models to constrained or on-device targets.
Pay
Zone A: $218,400 - $283,900 USD base salary; Zone B: $194,100 - $252,300 USD base salary; Zone C: $172,200 - $223,900 USD base salary
Tools
Python, PyTorch, TensorRT, ONNX Runtime, CoreML, TFLite, FlashAttention, Stable Diffusion
AI focus
Develops and deploys computer-vision and multimodal AI, including generative world models, neural rendering, image/video models, and reasoning over images, text, and 3D inputs.
Model-written summary · the original posting is authoritative