Real-Time Video AI Research Engineer

Google; Greater London; On-site; GBP 140,000 - 180,000; Full time

Job summary

Google is seeking a research-focused Software Engineer to design, implement, and scale diffusion models for real-time video generation. You will collaborate with research and product teams to prototype architectures, run large-scale tests, and deploy promising ideas broadly across the company.

You will lead high-performance GPU/TPU kernel development, optimize inference, and interface with labs worldwide to push frontier AI while upholding safety and ethics.

Qualifications

• Bachelor’s degree or equivalent practical experience.
• 5 years of experience developing, training, and deploying diffusion models and generative video architectures.
• Experience writing custom GPU/TPU kernels for model serving.

Responsibilities

• Architect and implement next-generation model architectures in collaboration with research teams to advance frontier world models.
• Design and iterate on real-world evaluation pipelines to systematically measure and accelerate model capabilities.
• Engineer high-performance model distillation and optimization techniques to support real-time serving at global scale.
• Build scalable data ingestion pipelines and automated filtering mechanisms, using rigorous dataset analysis to maximize training quality and drive downstream benchmark gains.

Skills

Diffusion models

Generative video architectures

GPU/TPU kernel development

JAX or PyTorch

Education

Bachelor's degree or equivalent practical experience

PhD in CS/ML/AI or related field

Tools

JAX

PyTorch

Job description

Google is seeking a research-focused Software Engineer to design, implement, and scale diffusion models for real-time video generation. You will collaborate with research and product teams to prototype architectures, run large-scale tests, and deploy promising ideas broadly across the company.

You will lead high-performance GPU/TPU kernel development, optimize inference, and interface with labs worldwide to push frontier AI while upholding safety and ethics.