Staff Machine Learning Engineer

Job overview

The Staff Machine Learning Engineer role is an onsite position based in Mountain View, CA. Compensation is not specified in the posting. This role supports the On-Device & Mobile AI team. This role exists to optimize and deploy state-of-the-art multi-modal models on mobile and constrained hardware. You will own the inference stack, ensuring that generative models run fast, small, and reliably within a browser-native runtime, directly shaping the AI features experienced by billions of players.

What you'll do

• Own the optimization pipeline for models, including export, quantization, and kernel-level tuning.
• Perform low-level performance work using WebGPU and native kernels.
• Integrate ML runtimes with the game engine.
• Partner with research scientists to productionize novel architectures.
• Lead and mentor a team of engineers.

What we're looking for

Skills & competencies

full-time

day shift

mountain view ca

unity

python

webgpu

machine learning

on-device ai

mobile ai

performance engineering

Work arrangement

• Weekend coverage required

Benefits & perks

• Comprehensive health, life, and disability insurance, commute subsidy, employee stock ownership, retirement plans, generous vacation, and mental health programs.

Why this role

Focus on on-device AI inference and performance optimization for mobile.

About the employer

Unity Technologies is hiring for this role. Industry: Periodical Publishers. Sector: 51.