Staff Machine Learning Engineer

Unity Software
Mountain View
Workplace: OnsiteFull timeUSD 218,400 - 327,600 annuallyFunction: Data Science & Machine LearningExperience: 6+ yearsSkills: ["Technical leadership","Mentorship","Cross-functional collaboration","Measurement and benchmarking"]

Build and productionize AI-driven game experiences using generative world models, neural rendering, and multimodal understanding. Lead technical vision and roadmap for transformer, diffusion, vision-language, and JEPA-style architectures, translating CV research into robust training, deployment, and evaluation pipelines. Mentor ML engineers, partner with research and platform teams, and optimize quality, latency, memory, and cost across cloud GPUs and on-device inference.

Loading

Loading job details...

Preparing the role view and application actions.

FursaFursa
Unity Software
Unity Software
2 months ago

Staff Machine Learning Engineer

✓ Verified Job

Canonical indexed version, validated from employer's careers page.

Source: Company careers pageValidated by: Fursa AI
Last checked: 59 minutes agoStatus: Live

Job Summary

Build and productionize AI-driven game experiences using generative world models, neural rendering, and multimodal understanding. Lead technical vision and roadmap for transformer, diffusion, vision-language, and JEPA-style architectures, translating CV research into robust training, deployment, and evaluation pipelines. Mentor ML engineers, partner with research and platform teams, and optimize quality, latency, memory, and cost across cloud GPUs and on-device inference.
Location: Mountain View
Workplace: Onsite
Employment Type: Full time
Job Function: Data Science & Machine Learning
Seniority: Mid level

Key Responsibilities

  • •Set the technical vision and roadmap for computer vision and multimodal AI models across transformers, diffusion, vision-language, and JEPA-style architectures.
  • •Design and implement models for image/video understanding and generation, including segmentation, detection, dense prediction, and multimodal reasoning over images, text, and 3D inputs.
  • •Own the path from research prototypes to production systems, covering training, fine-tuning, distillation, export, and serving across cloud GPUs and on-device inference when needed.
  • •Collaborate with research scientists to translate novel CV and multimodal architectures into deployable, well-engineered implementations.
  • •Lead and mentor ML engineers; define engineering best practices, benchmarking/evaluation methodology, and KPIs for quality, accuracy, latency, memory, and cost.

Pay and Benefits

Salary: USD 218,400 - 327,600 annually
Equity and Bonus:Equity
Perks:Health InsuranceLife InsuranceDisability InsuranceEquityCommuter Benefits

Key Requirements

  • •6+ years in ML engineering with deep experience in computer vision and/or multimodal modeling.
  • •Production experience with transformer- and diffusion-based vision models (e.g., ViT, CLIP/SigLIP, Stable Diffusion, DETR/SAM-style).
  • •Strong end-to-end model lifecycle skills: data curation, training/fine-tuning, evaluation, and large-scale serving.
  • •Familiarity with efficient attention, diffusion samplers, multimodal fusion, and vision-language alignment techniques.
  • •Strong Python and PyTorch skills with solid software engineering fundamentals.
Experience:6+ yearsComputer visionMultimodal AIDeep learningTransformersDiffusion models
Skills:Technical leadershipMentorshipCross-functional collaborationMeasurement and benchmarking
Languages:English
Tech Stack:PythonPyTorchTransformersDiffusion modelsVision-language modelsFlashAttentionTensorRTONNX RuntimeCoreMLTFLiteONNXMLIRTVMXLAQuantizationPruningKnowledge distillation

Eligibility

Work Authorization:Authorization required. Sponsorship not provided.

Company Brief

Unity Software
Develops the Unity real-time 3D development platform used to create, operate, and monetize interactive experiences and games across mobile, console, PC, and AR/VR platforms. Serves developers, studios, and enterprises worldwide.
Industry: Developer Tools
Company Size: Enterprise (1,001+ employees)
Revenue: USD 1B+
Growth: Public Company
Valuation: Public Company (Market Cap in USD)
Funding: IPO / Publicly Listed
Headquarters: San Francisco, United States
Founded: 2004
Glassdoor
Glassdoor: 3.8
WebsiteLinkedIn