Senior AI Engineer

Senior AI Engineer | Enterprise AI Grounding & RAG Systems | Stockholm

We're working with a well-funded B2B SaaS company that has spent the last decade building a market-leading platform used by global enterprise customers across technology, manufacturing, healthcare, financial services and beyond.

They've recently launched a small, highly strategic AI innovation team focused on solving one of the biggest challenges facing enterprise AI today:

How do organisations ensure AI systems produce reliable answers based on trusted company knowledge?

This team is building the infrastructure layer that allows enterprise AI systems to ground responses in structured, governed content rather than relying on unverified information. It's a genuine R&D environment operating with startup autonomy but backed by an established product company and customer base.

The Opportunity

They're looking for a Senior AI Engineer to take ownership of the retrieval and knowledge layer behind next-generation enterprise AI products.

This is not a role for someone who has only experimented with AI tooling or built basic chatbot applications.

They're looking for someone who has successfully shipped AI systems into production and understands the challenges around retrieval quality, hallucinations, evaluation, scalability, and real-world customer adoption.

You'll work on problems including:

• Knowledge graphs and semantic retrieval
• Enterprise RAG architectures
• Advanced search and retrieval systems
• Vector databases and embeddings
• Agentic workflows and AI orchestration
• Evaluation frameworks and AI quality measurement
• Scaling AI products from prototype to production

What You'll Be Doing

• Designing and building knowledge graph architectures
• Owning retrieval pipelines end-to-end
• Improving retrieval quality, latency and AI accuracy
• Collaborating directly with customers on real-world AI challenges
• Building production-grade AI systems used by enterprise organisations
• Influencing technical strategy and architecture decisions

Technical Background

We're particularly interested in engineers with experience across several of the following:

• RAG and retrieval systems
• Knowledge graphs
• Graph databases
• Vector databases
• Semantic search
• Agentic AI systems
• AI evaluation frameworks
• Python
• TypeScript
• AWS or other cloud platforms
• Infrastructure as Code
• Containerisation

Experience with technologies such as Neo4j, Neptune, RDF, SPARQL, LangGraph, AutoGen or similar would be beneficial.

Who This Could Suit

• Senior AI Engineers
• Applied AI Engineers
• AI Platform Engineers
• ML Engineers
• Staff Software Engineers working in AI
• Engineers building production RAG systems
• Search & Retrieval Engineers
• Knowledge Graph Engineers

Why It's Interesting

• Small, highly autonomous team
• Significant ownership and technical influence
• Greenfield AI products
• Complex engineering problems rather than AI demos
• Direct access to company leadership
• Enterprise-scale challenges and customers
• Long-term product vision with strong backing
• Opportunity to help define how enterprises build reliable AI systems

The company operates a hybrid model in Stockholm and can consider English-speaking candidates already based in Sweden.

If you'd like to hear more, feel free to reach out for a confidential discussion.