About the role
You will lead AI initiatives by researching, prototyping, and deploying models to production to improve math learning outcomes. You will work closely with product and engineering teams to build AI agents and optimize systems based on classroom feedback.
What they look for
Requirements
The role requires strong technical skills in Python, SQL, and AI development tools with a focus on hands-on coding. Candidates must be output-oriented, ambitious, and capable of working in a fast-paced startup environment.
Benefits
Full description
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 systems
- Building AI agents and agentic workflows
- Prototyping quickly and iterating based on real classroom feedback and measurable impact
- Deploying and monitoring models in production, improving quality, speed and reliability over time
- Leverage AI to scale both the product and the business
- Working closely with the team to define what “great” looks like for AI in education, and pushing the product forward fast
- This 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 impact
- Have a big interest in AI, stay up-to-date with the latest releases and enjoy experimenting
- Be very ambitious and like chasing challenging targets
- Be extremely output-oriented and find satisfaction in driving projects to completion
- Eager to learn new things, and have experience of building stuff
- Great at communicating
- Be hardworking and willing to join a fast-paced and rapidly growing startup
Requirements
- Coding skills that enable you to build stuff
- Strong technical skills
- Hands-on experience vibe coding with AI tools
- Fluent in English
Our tech stack
- Languages: Python, SQL, JavaScript/TypeScript
- AI & LLMs: multiple LLM providers, Langfuse for evaluation and observability
- Infrastructure: AWS
- AI 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 goals
- Fantastic colleagues in a fun environment with great opportunities to learn develop your skills
- Competitive salary and benefits
- An opportunity to work on a meaningful mission and have an exceptional impact on education, society, and how children learn math across the world
- Travel 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.
#LI-MAGMA
Similar roles
-
Jr Data Scientist (Fixed Term)
NielsenIQ Ciudad de México, Mexico
-
Product Data Scientist (Client-facing)
Similarweb Tokyo, Japan
-
Data Scientist III
Conduent Noida, Uttar Pradesh, India
-
Senior Data Scientist
team.blue Global Ghent, Flemish Region, Belgium
-
Senior Applied Data Scientist
dunnhumby London, England, United Kingdom
-
Senior Data Scientist
Publicis Groupe Holdings B.V Bengaluru, Karnataka, India