Data Scientist
TL;DR
We are looking for a hands-on AI Engineer / Data Scientist who loves building and shipping features. You’ll take big ownership of Magma’s AI projects, prototype fast, deploy models to production, and help improve math learning for millions of students. Stockholm-based, full-time, 5 days in office.
About Magma Math
Magma is one of the world’s fastest-growing and most celebrated edtech companies, transforming math education through the power of AI. Our platform turns handwritten solutions into digital insights, empowering teachers to focus on creativity and deeper learning while giving them real-time visibility into student progress.
The result? More inclusive, engaging, and effective math classrooms for all learners.
But we’re not just building better tools. We’re tackling one of the biggest challenges in global education: math underperformance. With 1 in 5 students globally struggling in math, a subject that forms the backbone of so many career paths, we believe the opportunity to improve outcomes at scale is massive.
We’re already the go-to solution in thousands of schools and we’re just getting started. As we expand rapidly across the U.S. and Europe, we’re building a team as ambitious as our mission: to help millions of students unlock their full potential in math.
The Role
We are looking for builders that want to work with large data sets, build AI-powered products and ship things that make a real difference for teachers and students. In this role you will take large ownership of Magma’s AI initiatives, from research and prototyping to production and iteration, working together with product and engineering.
Your Work Will Span Across Areas Such As
Digging into data, building models, pipelines and AI-powered systemsBuilding AI agents and agentic workflowsPrototyping quickly and iterating based on real classroom feedback and measurable impactDeploying and monitoring models in production, improving quality, speed and reliability over timeLeverage AI to scale both the product and the businessWorking closely with the team to define what “great” looks like for AI in education, and pushing the product forward fastThis is a high-ownership role where you’ll be expected to move quickly, explore new ideas, and turn the best ones into real product improvements that are used by students every day
Who You Are And What You Know
We are searching for a team member who wants to take large ownership of Magma's different AI projects. To succeed you need to:
Be passionate about building stuff with big impactHave a big interest in AI, stay up-to-date with the latest releases and enjoy experimentingBe very ambitious and like chasing challenging targetsBe extremely output-oriented and find satisfaction in driving projects to completionEager to learn new things, and have experience of building stuffGreat at communicatingBe hardworking and willing to join a fast-paced and rapidly growing startup
Requirements
Coding skills that enable you to build stuffStrong technical skillsHands-on experience vibe coding with AI toolsFluent in English
Our tech stack
Languages: Python, SQL, JavaScript/TypeScriptAI & LLMs: multiple LLM providers, Langfuse for evaluation and observabilityInfrastructure: AWSAI development: Claude Code, Cursor and other AI-native development tools
Our stack evolves quickly and we care more about using the right tools and shipping great systems than any specific framework.
What We Offer
Play a key role in an exciting journey. We are a young company with ambitious targets and meaningful goalsFantastic colleagues in a fun environment with great opportunities to learn develop your skillsCompetitive salary and benefitsAn opportunity to work on a meaningful mission and have an exceptional impact on education, society, and how children learn math across the worldTravel and meet users and colleagues around the world
Location: Stockholm, Sweden.
Note: We practice a 5 day in office policy, and we look forward to working together closely with you :)
Our Recruitment Process
A call with the Hiring Manager.
1st interview with Hiring Manager and People team member.
2nd interview with Hiring Manager and Magma team member.
3rd interview with CPO and Magma team member.