Principal Machine Learning Engineer

Unity Software
Mountain View
Workplace: OnsiteFull timeUSD 278,100 - 417,100 annuallyFunction: Data Science & Machine LearningSkills: ["Technical leadership","Mentorship","Cross-functional collaboration","Measurement-driven improvement","Performance focus"]

Build on-device AI inference for a browser-native game runtime, helping state-of-the-art multimodal transformer and diffusion models run fast, small, and reliably inside production. Own the end-to-end optimization and integration stack: export, graph transformation, operator fusion, kernel tuning, and runtime scheduling. Lead engineering standards, mentor engineers, and partner across research and platform teams to meet tight latency, memory, and power budgets across mobile NPUs and GPUs and desktop runtimes.

Loading

Loading job details...

Preparing the role view and application actions.

FursaFursa
Unity Software
Unity Software
2 months ago

Principal Machine Learning Engineer

✓ Verified Job

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

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

Job Summary

Build on-device AI inference for a browser-native game runtime, helping state-of-the-art multimodal transformer and diffusion models run fast, small, and reliably inside production. Own the end-to-end optimization and integration stack: export, graph transformation, operator fusion, kernel tuning, and runtime scheduling. Lead engineering standards, mentor engineers, and partner across research and platform teams to meet tight latency, memory, and power budgets across mobile NPUs and GPUs and desktop runtimes.
Location: Mountain View
Workplace: Onsite
Employment Type: Full time
Job Function: Data Science & Machine Learning
Seniority: Sr. Manager level

Key Responsibilities

  • •Own the end-to-end on-device inference and integration stack, from model export and optimization through shipped engine features running at interactive frame rates.
  • •Build and drive the optimization pipeline (graph transformation, operator fusion, memory-layout planning, and hardware-specific kernel tuning) across NPU, mobile GPU, and desktop/laptop GPU.
  • •Make authoritative performance trade-offs using quantization, weight sharing, pruning, and knowledge distillation while validating against latency, memory, power, and quality targets.
  • •Write and tune WebGPU compute shaders and native kernels as needed; profile with browser/platform tools to eliminate op- and memory-bandwidth bottlenecks.
  • •Lead and mentor a team, set engineering best practices and performance regression gates, define KPIs, and partner with research, platform engineers, and product managers.

Pay and Benefits

Salary: USD 278,100 - 417,100 annually
Equity and Bonus:Equity
Perks:Health InsuranceLife InsuranceDisability InsuranceCommuter BenefitsRetirementPaid Leave

Key Requirements

  • •8+ years in software/ML engineering with at least 4 years focused on on-device/edge inference or real-time performance-critical systems.
  • •Proven production deployment of transformer and/or diffusion models shipped on mobile, desktop, or embedded hardware.
  • •Hands-on experience deploying models through WebGPU (e.g., ONNX Runtime Web, Transformers.js, WebLLM, TensorFlow.js) including WGSL compute shader work and WebGPU limits/bindings.
  • •Deep expertise with an inference runtime (ONNX Runtime/ORT Web, CoreML, TFLite, ExecuTorch) plus operator fusion, memory layout, and runtime scheduling knowledge.
  • •Low-level performance engineering with GPU/compute APIs (WebGPU/WGSL, Metal, Vulkan, D3D12, or CUDA) and profiling tools to identify time and memory bottlenecks.
Experience:On-device AIEdge inferenceReal-time systemsTransformersDiffusion models
Skills:Technical leadershipMentorshipCross-functional collaborationMeasurement-driven improvementPerformance focus
Languages:English
Tech Stack:WebGPUWebNNWGSLTypeScriptJavaScriptONNX Runtime WebTransformers.jsWebLLMTensorFlow.jsCoreMLONNX RuntimeTFLiteExecuTorchMetalVulkanSPIR-VD3D12CUDAChromeDawn GPU traces

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