AI Engineer

Umain AB; Stockholms kommun; On-site; SEK 500,000 - 700,000; Full time

Benefits offered by this job

Work-life balance

Personal coaching

Diverse projects with global brands

Fair work environment

Job summary

Umain AB is seeking a dedicated developer to assist clients in AI solutions and deliver innovative products. Responsibilities include architecting AI solutions and maintaining data workflows. The ideal candidate has strong software engineering experience and an understanding of AI concepts. Umain offers a fun, fair work environment, opportunities for personal coaching, and a transparent career framework supporting work-life balance. Join a team passionate about technology and eager to innovate.

Qualifications

• Strong background in software engineering and product development.
• Experience in AI, particularly with agents and context engineering.
• Ability to work effectively in collaborative product/engineering environments.

Responsibilities

• Assist clients in determining the suitability of AI solutions.
• Architect and implement agentic solutions using various frameworks.
• Develop and maintain data pipelines and machine learning workflows.

Skills

Software engineering

AI concepts understanding

Product development experience

End-to-end solution delivery

Tools

Mastra

Microsoft Agent Framework

LangChain/LangGraph

Langfuse

Promptfoo

Job description

Responsibilities

• Assisting our clients in clarifying their questions and determining if using AI is a suitable solution for them
• You understand 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

Qualifications

• 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

Ideal Candidate

• Great delivery skills: Proven capability to deliver AI solutions
• Strong AI intuition: A deep understanding of AI concepts
• Curiosity: Enthusiasm for emerging technologies, creativity in problem‑solving, and an 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
• You're a developer with a passion for innovation and a knack for problem‑solving.
• You're eager to learn, grow, and take on new challenges.
• You value collaboration, diversity, and the opportunity to make a significant impact.
• You believe in the power of technology to shape the future.

Benefits

• 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.
• Career Framework: We believe in fairness and clarity, which is why we've established a transparent career framework to ensure equal pay and equal opportunities for all.
• Work‑Life Balance: At Umain, diversity is our strength. We understand that life doesn't pause when you walk into the office. That's why we've cultivated an environment that supports an excellent work‑life balance, respecting your needs whether it's taking care of your family, your pets, or yourself.
• Innovations and Learning: Technology is always evolving, bringing new challenges and opportunities for collaboration and learning. Our developers work closely with designers, analysts, testers and more to tackle these challenges head‑on, sharing knowledge and growing together.
• Skills Over Degrees: We're more interested in what you can do than in the diplomas you hold. Talent, drive, and creativity are what count at Umain. Here, you'll find opportunities to grow, learn, and excel in your career, regardless of your educational background.

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.