ML Engineer, Backend

Prior Labs GmbH; North Merrick (NY); Hybrid; USD 160,000 - 210,000; Full time

Job summary

Prior Labs GmbH is seeking an experienced backend infrastructure engineer to own the inference serving path and build scalable production systems for TabPFN-era foundation models. You will shape the core backend, design robust APIs, and manage cloud infrastructure with a strong emphasis on reliability and cost-efficiency.

You will collaborate with AI researchers to translate cutting-edge models into scalable production systems, with opportunities to specialize in inference optimization, backend

Qualifications

• 3+ years of professional experience in machine learning and backend engineering, with a proven track record of managing production infrastructure.
• Proven experience deploying and operating machine learning models in production, with strong understanding of model behavior under real traffic.
• Strong, hands-on experience designing and building production-grade APIs and backend services.
• Significant experience with cloud platforms, with GCP preferred.
• Strong experience with Infrastructure as Code (IaC) using Terraform.
• Significant experience with containerization and orchestration (Docker, Kubernetes).
• Proficiency in Python.

Responsibilities

• Own the serving path end-to-end: latency, throughput, batching, memory behavior on large inputs, and cost per prediction.
• Architect and design robust, scalable backend systems and production-grade APIs for serving foundation models.
• Build and implement high-quality, maintainable core backend services.
• Own infrastructure design, deployment, and management on cloud platforms with focus on reliability, monitoring, observability, and cost-efficiency.
• Ensure secure, GDPR-compliant data handling, including storage, access control, usage tracking, and quota management.
• Champion best practices for testing, CI/CD, documentation, and security within the engineering team.

Skills

ML backend infra

Python

Terraform

Docker

Kubernetes

GCP

API design

CI/CD

Tools

PyTorch

FastAPI

Git

Job description

Who we are

Foundation models transformed text and images. Structured data - the largest and most consequential data format in the world - stayed untouched, until now. What LLMs did for language, we're doing for tables.

We pioneered tabular foundation models: TabPFN v2 was a Nature cover story, has passed 3.5M+ downloads and 7,500+ GitHub stars, and runs in production from detecting lung disease with Oxford Cancer Analytics to preventing train failures with Hitachi. The hardest problems - millions of rows, real-time inference, entirely new modalities - are still open, and no one else is working on them at this level.

We're a small, highly selective team of 40+ with backgrounds from Google, DeepMind, Meta, Apple, Amazon, Jane Street, and CERN, led by Frank Hutter, Noah Hollmann, and Sauraj Gambhir, and advised by Bernhard Schölkopf and Turing Award winner Yann LeCun.

In July 2026, less than 18 months after our €9M pre-seed, we joined SAP as an independent frontier AI lab - same team, mission, and open-weights models, now backed by more than €1 billion over four years.

About the role

You will have ownership over designing, building, and scaling the core systems that bring Prior Labs' foundation models to the world. This is a unique opportunity to make fundamental architectural decisions, establish engineering best practices from the ground up, and profoundly shape the technical direction for serving state-of-the-art AI.

TabPFN does its learning in the forward pass, and it now handles datasets in the millions of rows. Making that fast, reliable, and affordable is an exciting engineering challenge.

You'll work directly with world-class AI researchers, translating cutting-edge models into scalable production systems. This role offers significant autonomy and impact, with clear paths to specialize in areas you're passionate about (like inference optimization and core backend systems) or grow into a technical leadership position as our team expands. You won't just be implementing features; you'll be building the backbone of our company.

What you'll work on:

• Own Inference: Own the serving path end-to-end: latency, throughput, batching, memory behavior on large inputs, and cost per prediction. Our models serve through our managed API, inside customer VPCs, and as self-hosted open weights, and you’ll have an impact across the board.

• Architect & Design: Design robust, scalable, and secure backend systems and production-grade APIs for serving our foundation models.

• Build & Implement: Develop high-quality, maintainable code for core backend services.

• Own Infrastructure: Design, deploy, and manage core infrastructure on cloud platforms, focusing on reliability, monitoring, observability, and cost-efficiency.

• Ensure Compliance & Security: Implement secure, GDPR-compliant systems, including data storage, access control, usage tracking, and quota management.

• Champion Best Practices: Drive high standards for testing, CI/CD, documentation, and security within the engineering team.

You may be a good fit if you have:

• 3+ years of professional experience in machine learning and backend engineering, with a proven track record of managing production infrastructure.

• Proven experience deploying and operating machine learning models in production, with a strong understanding of how models behave and fail under real traffic.

• Strong, hands‑on experience designing and building production‑grade APIs and backend services.

• Significant experience building and operating services on cloud platforms, with GCP preferred.

• Strong, hands‑on experience with Infrastructure as Code (IaC) using tools like Terraform.

• Significant experience with containerization and orchestration technologies (Docker, Kubernetes).

• Proficiency in Python.

What sets you apart

• Experience building or managing infrastructure specifically for machine learning (MLOps, model serving frameworks, feature stores, data pipelines).

• Deep experience with inference optimization (quantization, compilation, dynamic batching, caching strategies, interplay with distributed computing).

• Experience with PyTorch and FastAPI specifically.

• Contributions to relevant open‑source projects.

Life at Prior Labs

You'll work alongside researchers and builders who hold themselves to a very high bar - in the quality of their work and in how they work with each other. We move fast and still take the time to do things right.

Our teams are based in Berlin, Freiburg, and New York - when you're working on something as hard as TabPFN, being in the same room matters. But great people come from everywhere, and in exceptional cases we're open to remote, which usually means frequent travel to one of our offices. Wherever you're based, the whole company comes together regularly for offsites to build and celebrate together.

Our Commitments

The best products and teams are built by people with a wide range of perspectives and backgrounds. We welcome applications from all identities and walks of life - especially if you've ever felt discouraged by \"not checking every box\" - and provide equal opportunities regardless of gender, sexual orientation, origin, disability, or any other trait that makes you who you are.

We care about how your data is handled - see our Recruiting Data Privacy page