Prior Labs logo

Research Engineer, Foundation Model

Prior Labs·Salary not specified

Primary stack

PythonPyTorch

Job description

About the Position

Research Engineer, Foundation Model


About the Role

At Prior Labs, we are revolutionizing the way we handle structured data. Our pioneering work with tabular foundation models has already made significant strides, but the hardest problems are still ahead. As a Research Engineer, you will be at the forefront of this exciting field, designing experiments, contributing to papers, and writing the code that turns architectural ideas into trained models.

Key Responsibilities

  • Design and test novel architectures
  • Run ablations and analyze scaling behavior
  • Write training and evaluation infrastructure for rapid experimentation
  • Contribute to the development of efficient and scalable models for real-world deployment

The Problems We're Solving

  • Scaling transformer architectures from 10K to 1M+ samples without the structural assumptions that make language models scale
  • Building multimodal models that combine tabular, text, and numerical understanding
  • Making models efficient enough for real-world deployment
  • Designing architectures for time series, forecasting, anomaly detection, and multiple related tables

Requirements

  • Master's or PhD in Computer Science or a related field, plus 3+ years of experience building ML systems in research or industry
  • Publications at top ML venues (NeurIPS, ICML, ICLR, etc.) or equivalent demonstrated research impact
  • Deep proficiency in Python, PyTorch, and the broader ML and data science ecosystem (scikit-learn, pandas, NumPy)
  • Experience implementing and training neural network architectures, ideally transformers or foundation models
  • Solid understanding of training dynamics, scaling behavior, and common failure modes in deep learning systems
  • Genuine interest in model efficiency - making large models faster, more scalable, and practical to deploy

Nice to Have

  • Experience at an early-stage startup or as a founding engineer
  • Contributions to open-source ML libraries or tools
  • Experience with model distillation, inference optimization, or on-device ML
  • Background in tabular data, time series, or other structured data

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


About the Company

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.

© Prior Labs. This job description was sourced from the employer's public career page. TheJob is not the employer — we index the posting and route candidates to the source. All content rights and hiring decisions belong to the employer.

More at Prior Labs

All 22 roles

Popular searches