Machine Learning Engineer
OrbDB Labs AB
Stockholm, Stockholms län
Onsite
2026-07-05
Estimated salary · Stockholm
~ SEK 572,400 - SEK 800,400
Low
SEK 572K
Median
SEK 674K
High
SEK 800K
Market in Stockholm · SCB 2025
Estimated net pay
SEK 37,053 - SEK 47,620
/month · 22% withheld
after tax & contributions · on the estimated salary · Individual taxation — marital status and dependents do not affect it
Job description
What we're building
OrbDB is building data infrastructure for AI reliability. For every prediction a model makes, the platform determines whether the model is sufficiently certain for the result to be acted on automatically or whether the case should be routed to a human reviewer.
Today’s AI production systems are unable to distinguish which of their predictions are trustworthy. We are building the layer that allows organizations to automate the cases where automation is statistically justified, and to escalate the rest with confidence.
OrbDB is founded and led by researchers with deep expertise in the underlying methods.
The role
You will work on the models that sit at the center of our platform. Our work is built around Graph Neural Networks, and the questions you will engage with are the ones that sit beneath the surface of any serious deep learning system: questions about architecture, training behaviour, optimization, and the relationship between what a model is doing and what we expect it to do.
This is a role for someone who knows the fundamentals of deep learning well enough to reason about them from first principles, not from tutorials. You will work closely with our research-led founding team, and the questions you take on will move between the practical and the foundational, often within the same week. Unlike other AI startups, OrbDB builds on a mathematical foundation. So do the teams behind it. OrbDB Labs is a place where solid ideas and good taste matter more than loud voices.
Specifically, you will:
Train, evaluate, and improve the models that power the platform.
Diagnose model behavior at a level deeper than metrics, and propose changes grounded in the underlying mathematics.
Make principled choices about model design as required.
Work alongside the engineering team to deliver research-grade models into a production system that customers can rely on.
What we are looking for
2-4 years of experience working with deep learning models in a serious