AI Engineer

Umain; Stockholms kommun; Hybrid; SEK 400,000 - 600,000; Full time

Benefits offered by this job

Professional development opportunities

Creative and fair work environment

Personal coaching

Job summary

Umain, a dynamic digital product consultancy located in Stockholm, is seeking an AI Engineer to build and deliver advanced AI solutions. In this role, you'll work closely with clients to clarify their AI needs and develop robust solutions while collaborating on diverse projects.

The ideal candidate possesses strong product engineering skills and a deep understanding of AI technologies. Join a talented team dedicated to shaping the future through innovative projects.

Qualifications

• Experience in AI-powered solutions and product development.
• Familiarity with architecting agentic solutions.
• Passion for learning and adapting to evolving technologies.

Responsibilities

• Assist clients in determining suitable AI solutions.
• Develop and maintain AI solutions with a product engineering mindset.
• Collaborate with cross-functional teams to deliver AI solutions.

Skills

AI solutions delivery

Understanding core AI concepts

Product development

Stochastic test casing

End-to-end data pipelines

Tools

Mastra

Microsoft Agent Framework

LangChain/LangGraph

Langfuse

Promptfoo

Job description

At Umain, we're not just creating software; we're crafting the future, one innovative project at a time. With our motto "Shape, Ship, Scale," we empower our team to not only envision the future but also play a pivotal role in building it. We're a dynamic team where you can shape the future, ship groundbreaking solutions, and scale your abilities in many different ways.

We're on the lookout for developers ready to join us in this journey, bringing passion, creativity, and dedication along with them. The role is ideal for engineers with demonstrable skills in AI as well as for software engineers who have recently transitioned to AI and ML expertise, and offers an opportunity to develop your craft whilst working closely with clients on delivering AI solutions to real-world challenges.

We are seeking an AI Engineer with experience in building robust, scalable AI powered solutions. The ideal candidate combines a product engineering mindset with a deep enthusiasm for advancing AI capabilities. They demonstrate a great understanding of core AI concepts, product development workflows, and embody team spirit.

As an AI Engineer at Umain, your role will involve:

• Assisting our clients in clarifying their questions and determining if using AI is a suitable solution for them
• Understanding the limitations of a model and how to address them within the product's UX
• Experience architecting agentic solutions (context, memory, and tool/skill handling) and implementing them using frameworks such as Mastra, Microsoft Agent Framework, LangChain/LangGraph
• Applying stochastic test casing and trace-based reasoning to evaluate, test, and improve agent behaviour, supported by observability tooling (Langfuse, Promptfoo, etc.)
• Understanding and applying core AI concepts such as redlining, document chunking for accuracy, efficient use of embeddings, and debugging AI systems
• Validating, deploying, monitoring, and maintaining built models, ensuring they can be used at scale
• Developing and maintaining end-to-end data pipelines and machine learning workflows, optimizing for performance and reliability
• Working with diverse, cross-functional teams on projects that deliver groundbreaking AI solutions for global brands

We also welcome experienced software engineers looking to transition into AI:

• Strong product development background with several years of software engineering experience
• Active engagement in AI (agents, wrappers, context engineering)
• Knows how to work efficiently in a product/engineering team

Who You Are

• Great delivery skills: Proven capability to deliver AI solutions
• Strong AI intuition: Deep understanding of AI concepts
• Curiosity: Enthusiasm for emerging technologies, creativity in problem-solving, and eagerness to learn in a rapidly evolving AI landscape
• Full-stack product thinking: Frontend or backend experience outside the traditional AI field, with the ability to build end-to-end solutions
• Developer with a passion for innovation and a knack for problem-solving
• Eager to learn, grow, and take on new challenges
• Values collaboration, diversity, and the opportunity to make a significant impact
• Believes in the power of technology to shape the future

What We Offer

• A chance to work closely with global brands on a range of diverse and unique projects that explore and truly push the boundaries of what's possible
• A fun and fair work environment where you have the opportunity to expand your skills through unique challenges and fantastic colleagues
• A personal coach and a setting where you can grow professionally and personally
• Being part of multidisciplinary teams, participating in agile teams, being an important player in a highly ambitious setting

Background Check: Umain conducts a background check for final candidates in accordance with our internal policies. The check is carried out only with your consent and is ordered by our CISO through our approved screening partner. The scope of the check depends on the role level. This step is performed late in the recruitment process and is used solely to support a secure and fact-based hiring decision.

Umain is a digital product consultancy. We shape, ship and scale digital products at speed. Founded in Stockholm in 2001, we are a growing team of 100+ experts in digital product strategy, design, and technology. We are a part of Eidra, a group of top-notch companies specializing in strategy, creativity, and engineering. As a collective we guide leaders in making meaningful and lasting change.