ML Inference Engineer

Description

SpAItial 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.