Machine Learning Inference Engineer

About SpAItial AI

SpAItial AI is at the forefront of developing advanced World Models that extend beyond traditional 2D imagery, integrating generative AI, computer vision, and physics-based simulation to comprehensively understand three-dimensional environments. Their work impacts various sectors, including robotics, augmented reality/virtual reality, gaming, and cinema, by generating immersive, physically grounded 3D worlds.

Role Overview

We are looking for a Research Engineer to take full ownership of the backend systems powering our World Models' inference. Your responsibility will be to design, build, and maintain inference services and APIs that deliver cutting-edge research models as scalable, reliable, and efficient production systems. This hands-on position requires previous experience managing production AI APIs and working collaboratively with researchers to deploy new models effectively to end users.

Key Responsibilities

• Manage and support the inference APIs and services underlying our products, overseeing the full lifecycle from incoming requests to output generation.
• Enhance system performance by optimizing GPU utilization, reducing memory consumption, and increasing overall efficiency at scale.
• Integrate and apply contemporary methods to improve inference speed and responsiveness.
• Deploy research prototypes as stable production services, including configurability for customer-specific needs.
• Collaborate closely with research and product teams to transition new models into live environments accessible by users.

Required Qualifications

• Minimum of three years’ experience in software engineering within AI, machine learning, or computer vision product contexts.
• Proven expertise in designing and maintaining production APIs.
• Strong proficiency in Python programming and experience deploying containerized services optimized for GPU inference platforms.
• Hands-on experience analyzing and enhancing GPU inference performance, including profiling, memory management, batch processing, and addressing cold start challenges.
• Knowledge of 3D data structures or computer vision domains is considered an advantageous asset.

Diversity and Inclusion

SpAItial AI values diversity and inclusivity in the workplace. Applicants from all backgrounds and experiences are encouraged to apply. We are committed to ensuring equitable treatment throughout the hiring process.