Senior AI Engineer
Join us in our vision to help everyone in the world to find the right education!
We usually respond within a week
Are you passionate about building intelligent systems with cutting-edge AI? Do you want to lead the development of Generative AI applications that genuinely help students worldwide navigate their educational journey?
Keystone Education Group is seeking an experienced and innovative Senior AI Engineer to join our growing team. Based primarily in Sheffield, UK, but offering hybrid working from Sheffield or Stockholm, this is a fantastic opportunity to spearhead our initiatives in Generative AI and make a significant impact within a global EdTech leader.
As our Senior AI Engineer, you will be the technical lead for AI application development, focusing heavily on Generative AI. You'll own the end-to-end lifecycle – from ideation and architectural design through to deployment and iteration – of sophisticated AI solutions. This full-stack role requires not just deep AI expertise but also strong software engineering skills to build robust, scalable systems. You will collaborate closely with product managers, front-end developers, and DevOps engineers, guide technical direction, champion best practices, and stay at the forefront of the rapidly evolving AI landscape, particularly within the GenAI space.
We are looking for a seasoned AI/ML professional with a proven track record of building and deploying impactful AI solutions. You should possess strong analytical and problem-solving skills, combined with excellent communication and leadership potential.
Essential foundational skills include:
- Deep proficiency in Python or JavaScript/TypeScript and associated AI/ML SDKs.
- Solid experience with cloud platforms, specifically AWS (including services like SageMaker, Lambda, S3, EC2, Bedrock).
- Strong software engineering fundamentals (API design, data structures, algorithms, version control with Git, testing methodologies).
- Understanding of full-stack application architecture and integration points.
- Experience working in agile development environments and familiarity with CI/CD practices.
Our Technology Environment
You'll be working within a modern tech stack primarily centred around AWS. Our data strategy incorporates Knowledge Graphs, specifically using Neo4j. For large language models, we leverage APIs from providers like OpenAI and Google Gemini, while also valuing experience with deploying and managing open-source models. Our front-end stack includes React/Next.js, so familiarity here is beneficial.
Skills
Beyond the core competencies, success in this role hinges on specific, hands-on experience in the following areas:
- Generative AI Applications: Demonstrable experience designing, building, and deploying applications leveraging Generative AI techniques.
- RAG Pipelines: Deep understanding and practical experience in developing and optimising Retrieval-Augmented Generation (RAG) pipelines.
- GenAI Frameworks: Hands-on experience with key frameworks such as LangChain/LangGraph and/or LlamaIndex.
- LLM Integration: Proven ability to integrate effectively with LLM APIs (OpenAI, Gemini) and ideally experience handling open-source models.
- Knowledge Graphs for AI: Experience utilising Knowledge Graphs (Neo4j preferred) as part of AI architectures, as part of modern RAG systems.
- Highly Desirable: Experience building B2C applications (e.g., chatbots), exploring Agentic AI patterns.
What can we offer you?
- Freedom to help, plan and lead AI/ML architecture decisions in a fast-growing company.
- Extensive training and learning opportunities.
- Flexible hybrid work arrangements. An international work environment with colleagues from different nationalities.
- Autonomy and empowerment to pursue initiatives and ideas.
- Learn and educate through seminars at relevant conferences, or during our Hackathon events.
About the team:
The tech team works as an in-house tech department, based out in Stockholm, with the GenAI and Data team based in the UK. The responsibilities of the team encompass developing technical platforms and solutions to lay the ground for the rest of the business at Keystone Education Group. We are a rapidly growing company that is now looking for new talents to strengthen our team. Read more about us here: https://www.keg.com/ .
Do you want to join our family? Please send us your application by hitting the button "Apply here!". We are looking for someone who can start as soon as possible.
Keystone is an equal opportunity employer. We celebrate diversity and are deeply committed to fostering an inclusive environment for all employees.
- Business
- Keystone Education Group
- Role
- Tech/Product/Web
- Locations
- Sheffield - UK
- Employment type
- Full-time
- Remote Status
- Hybrid
- Employment level
- Senior level

Sheffield - UK
Three of many benefits
-
A fun workplace
We believe in creating a good environment between co-workers by participating in enjoyable activities such as sports events, after works and travelling. -
Opportunities to grow
We believe in giving every employee opportunities to develop their skills and capabilities by providing them with both external and internal courses and training. -
Reward for effort
We believe in showing appreciation to our employees, good performance could result in a trip with your team or in an individual prize or bonus.
Workplace & Culture
Our workplace culture is shaped by people who are driven to achieve, dedicated to making a difference — and having fun along the way.
At Keystone, we believe in expanding each individual’s potential, giving everyone the chance to grow, thrive, and be the best they can be. We craft innovative solutions and take a dynamic, flexible approach to problem-solving, embracing creativity as a catalyst for growth and progress.
We face challenges head-on, with the courage to explore new paths and the expertise to turn obstacles into opportunities. By connecting as a team, we foster a collaborative and supportive environment that brings out the best in each other and keeps us aligned around a shared vision.
With deep commitment to our mission, we deliver with focus and integrity - always working toward long-term success for the students, institutions, and families we serve.
Our workforce brings together talent from over 20 nationalities. What once felt “international” is now simply who we are - a truly global team where diversity strengthens our collaboration and drives us forward.
About Keystone Education Group
Keystone is a leader in marketing expertise for education worldwide. We help education providers to effectively reach their target audience. Keystone operates digital platforms for education search worldwide. Our expertly-crafted platforms and insight into the market enables us to continuously develop our search engines. Our ability to adapt to changing trends are making us the world’s leading and most popular search engines for education.
Senior AI Engineer
Join us in our vision to help everyone in the world to find the right education!
Loading application form
Already working at Keystone Education Group?
Let’s recruit together and find your next colleague.