Machine Learning Engineer - Stockholm

OrbDB Labs AB

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 technical setting, whether in research, industry, or a combination. If you are close to that range and the rest of the role fits, we would still like to hear from you. A real command of the fundamentals of deep learning. You should be comfortable reading a paper, implementing it, and reasoning about why a model is or is not behaving as expected. Strong engineering skills. You write code that others can build on, and you understand that a model is only useful once it runs reliably. Fluency with the modern deep learning toolchain, particularly PyTorch. Genuine interest in the statistical foundations of what we are building. Concepts like Conformal Prediction and calibration should be ones you are eager to understand deeply. Useful, but not required Experience with Graph Neural Networks specifically, or with the libraries that support them (PyTorch Geometric, DGL, or equivalent). A graduate degree in a quantitatively rigorous field, or equivalent depth acquired through other means. Open-source contributions in the ML or deep learning ecosystem, particularly to production-grade libraries. Experience moving models from research code into production systems. This is a Stockholm-based hybrid role. Candidates must be living in or willing to relocate to the Stockholm area before starting.

Nextory AB

You know that bit where every company says they’re “changing the world”? Yeah, it can feel a little forced. But we like to think our story actually has some realness to it. Our story begins with Shadi Bitar and Ninos Malki, who came to Sweden from Syria as kids and discovered that books could be a bridge - helping them learn the language, understand the culture, and explore new worlds. A thought emerged, what if everyone could have that? And just like that, Nextory was born - your all-access pass to read, listen, and discover new perspectives without limits. And now, we're on a quest to find a Machine Learning Engineer.  Let us tell you about the role 📖 As a Machine Learning Engineer at Nextory, you'll build the systems that power how users discover, understand, and engage with books. You'll work across AI agents, search, recommendations, retrieval, and product experiences, turning advances in AI into features that users interact with every day. Some examples of what you'll be doing: Build intelligent recommendation, search, and discovery systems that help users find their next favorite book and connect with the right content faster. Use AI, user behavior, and content understanding to continuously improve relevance, engagement, and user experience. Design, build, and operate scalable machine learning systems using LLMs, retrieval, recommendation, and agent technologies. Collaborate across product and engineering teams to deliver reliable AI-powered features with measurable user impact. Your story so far 👀 We believe you're an experienced engineer who enjoys solving complex problems and turning emerging technology into products that people love using. You are comfortable moving between product thinking and technical implementation, and you thrive in environments where experimentation, ownership, and fast iteration are part of everyday work. We also believe that you have: BSc in Computer Science or another relevant field, or equivalent work experience. At least 3 years of experience in a similar role Strong programming skills in Python. Experience working with AI agents and agent-based systems. Experience building and shipping LLM-powered features to production. The Nextory way of working 🏠 At Nextory, we believe a strong and inclusive culture is key to our success and that culture flourishes when we meet and collaborate in person as much as possible. That’s why we spend the majority of the working time at our beautiful office! We are based in Stockholm, Sweden, with offices in Paris and Bangalore. Our hybrid work policy includes four days in the office and Wednesdays as a work-from-home day. Your next story starts here 💫 We look forward to your application 💫. Applications are reviewed continuously, but please note that due to summer vacations, our response times may be a little longer than usual. If you have any questions about the role, please contact erika.thorslund@nextory.com. Please note that we do not offer relocation and that a six-month probationary period applies.

Vionlabs AB

About Vionlabs: Vionlabs is an AI company that helps media and entertainment businesses better understand and recommend video content. Our technology uses machine learning and computer vision to analyze content and improve user experiences for streaming platforms and media companies worldwide. About the Role: We are looking for a Machine Learning Engineer to join our growing team in Stockholm. As a Machine Learning Engineer, you will be responsible for developing, improving, and deploying machine learning models that power our AI-driven products. You will work closely with software engineers, data scientists, and product teams to build scalable solutions that deliver value to customers in the media and entertainment industry. Responsibilities: Design, develop, and maintain machine learning models and pipelines Analyze large datasets and extract actionable insights Build scalable data processing workflows Deploy and monitor machine learning models in production environments Collaborate with engineering and product teams to improve AI-powered services Continuously evaluate and improve model performance Qualifications: Bachelor's or Master's degree in Computer Science, Machine Learning, Data Science, Mathematics, or a related field Strong programming skills in Python Experience with machine learning frameworks such as PyTorch, TensorFlow, or similar Experience working with large datasets and data processing tools Knowledge of software engineering best practices Experience with cloud platforms such as AWS, Azure, or Google Cloud is a plus Strong analytical and problem-solving skills Excellent communication skills in English What We Offer: Opportunity to work with cutting-edge AI technology International and collaborative work environment Competitive compensation and benefits Flexible working arrangements Professional development opportunities Location: Stockholm, Sweden Employment Type: Full-time, permanent position.