ML Inference Engineer

United States Digital Space LLC; Greater London; On-site; GBP 80,000 - 120,000; Full time

Job summary

United States Digital Space LLC is seeking a Research Engineer to own the backend that serves world models. You will build and run the inference systems behind our product and API, turning frontier research models into fast, reliable services.

This is a hands‑on role for someone who has owned production APIs in an AI system and can collaborate with researchers to bring models to users. Responsibilities include owning inference services, optimizing GPU utilization, deploying models as production

Qualifications

• 3+ years of software engineering in an AI, machine learning, or computer vision product environment.
• Experience designing and owning production APIs.
• Strong Python, and experience shipping containerized services on GPU inference platforms.
• Hands‑on GPU inference performance work: profiling, memory, batching, and cold start.
• Familiarity with 3D or computer vision data is a strong plus.

Responsibilities

• Own the inference services and APIs that power our product, from request to generated output.
• Optimize serving performance, reliability, and cost at scale. Maximize GPU utilization and minimize device memory footprint.
• Apply latest techniques for improving inference performance.
• Deploy research models as production services, including customer‑specific configurations.
• Work with research and product colleagues to bring new models to users.

Skills

Production APIs

Python

Containerized services

GPU inference optimization

GPU performance profiling

Tools

Docker

Kubernetes

CUDA

Job description

the company is pioneering the next generation of World Models, pushing the boundaries of generative AI, computer vision, and the simulation of reality. We are moving beyond 2D pixels to build models that natively understand the physics and geometry of our world. Our mission is to redefine how industries, from robotics and AR/VR to gaming and cinema, generate and interact with physically-grounded 3D environments.

We’re seeking a Research Engineer to own the backend that serves our world models. You will build and run the inference systems behind our product and API, turning frontier research models into fast, reliable, and scalable services. This is a hands‑on role for someone who has already owned production APIs in an AI system and can work closely with researchers to bring new models to users.

Responsibilities

• Own the inference services and APIs that power our product, from request to generated output.
• Optimize serving performance, reliability, and cost at scale. Maximize GPU utilization and minimize device memory footprint.
• Apply latest techniques for improving inference performance.
• Deploy research models as production services, including customer‑specific configurations.
• Work with research and product colleagues to bring new models to users.

Key qualifications

• 3+ years of software engineering in an AI, machine learning, or computer vision product environment.
• Experience designing and owning production APIs.
• Strong Python, and experience shipping containerized services on GPU inference platforms.
• Hands‑on GPU inference performance work: profiling, memory, batching, and cold start.
• Familiarity with 3D or computer vision data is a strong plus.

At the company, we are committed to creating a diverse and inclusive workplace. We welcome applications from people of all backgrounds, experiences, and perspectives. We are an equal opportunity employer and ensure all candidates are treated fairly throughout the recruitment process.