AI Engineer

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, scaleable, 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 embodies 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

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

Devel