AI Engineer

Permanent
Technology
Experienced
Machine Learning, Machine Learning Engineer, Founding Engineer, ML Engineer, LLM, LLMs, Large Language models, NLP, Computer Vision
Stockholm
Blekinge
£70000 - £100000 per annum

We're looking for an ambitious AI Engineer with a focus on visual/generative AI to join our team.

Do you have a passion for turning research into practical applications? ✨

We are a fast-growing Stockholm-based startup with a team of 6, working on a cutting-edge creative product with AI at its core. We believe machine learning and AI are about to redefine what it means to be a creative, who can be part of a creative process and that creativity and AI are a uniquely powerful combination and an incredibly exciting field work in. We pride ourselves on our ability to rapidly adopt new developments, stay curious, and adapt quickly to changing circumstances.

In this role you will

Do R&D

  • Stay up to date with the latest research and advancements in the field of visual generative AI and machine learning.

  • Design, prototype, and evaluate new models and approaches that push the boundaries of what's possible.

  • Conduct experiments that drive product innovation and uncover novel solutions.

Build and launch

  • Implement state-of-the-art machine learning models with clean and efficient code.

  • Set up infrastructure and tools for fine-tuning and deploying models at scale.

  • Build and maintain data pipelines needed for your models.

  • Work closely with our engineering team to ensure seamless functionality and a high-quality user experience.

  • Monitor the performance of deployed models, making necessary adjustments to improve accuracy, efficiency and creativity.

  • Work closely with product managers, designers and engineers, to help shape the direction of our product.

In your first

Week

  • Set up your local development environment and dive into our codebase and existing model architectures.

  • Explore our product-experiment with generating art.

  • Deliver a small bug fix or quality-of-life improvement to production.

3 Months

  • Review the performance and retrain many of our production models.

  • Conduct experiments and iterate on your models, incorporating feedback from the team and customers.

  • Develop at least one novel approach to solve a customer need.

  • Set up monitoring and alerts for all critical AI infrastructure.

6 Months

  • Develop deep expertise in visual generative AI.

  • Optimize our model deployment and training infrastructure to drive enhanced performance while improving cost-efficiency.

  • Accelerate our R&D process to drive continuous product innovation.

You'll be a great fit if

Requirements

  • Bachelor's degree in Computer Science (Master's or Ph.D. in related fields preferred) or equivalent work experience.

  • 3+ years of hands-on experience with machine learning projects, focusing on visual/generative AI (computer vision, diffusion models, transformers).

  • Solid understanding of machine learning concepts and techniques, as well as their underlying mathematical principles.

  • Experience transforming research projects into scalable production services.

  • Proficiency in Python and strong overall programming skills.

Nice to have

  • Strong interest in creative applications of AI and a passion for pushing boundaries.

  • Contributions to AI research or applications in the form of publications or open-source projects.

  • Familiarity with advanced diffusion pipelines.

  • Experience deploying to GPU-accelerated cloud environments.

  • Prioritize delivering value to users over perfecting models.

What we offer

A challenging and fulfilling opportunity to contribute to the development of a groundbreaking creative product.

A supportive, collaborative, and innovative work environment with a team of like-minded individuals.

An opportunity to join an early-stage company and influence how we build the company before there are too many processes and structures in place.

Competitive salary and an attractive equity program in a start-up funded by top investors Creandum (1st investor Spotify, iZettle, depop) and EQT Ventures (Einride, Sonantic). 🦄

A learning environment, with interesting challenges to solve and an ambitious team to support you.

Office located in the heart of Stockholm.

Work in an organisation where AI is not an afterthought and a "side interest" but something that is there and has a seat at the table from the start.

Perks and Benefits

👩🏽‍💼

Business co-ownership

🏃🏽‍♀️

Fitness stipend

🤒

No qualifying days

👩‍🎓

Personal learning budget

🥳

Swap public holidays

🎂

Celebration

⛵️

6 weeks paid vacation

💻

The tools you need

Similar Jobs

£700 - £750 per day
West Yorkshire
We are seeking a highly skilled Technical Lead to join a major digital transformation programme within a large-scale public sector environment. This role will focus on leading engineering teams delivering high-profile, citizen-facing systems and shared digital capabilities used nationwide. As a hands-on technical leader, you will set direction for your team, ensure alignment with wider strategy, and drive the delivery of secure, reliable, and modern digital services.
£45000 - £55000 per annum
Essex
We're seeking a Mid-Level Full Stack Developer to join a growing Technology & Innovation team. You'll play a central role in building and enhancing web applications for both B2B and B2C clients. This position offers the chance to take greater ownership of projects, contribute to architectural discussions, and drive improvements across the stack.
£35000 - £45000 per annum
Essex
We're looking for a Junior Full Stack Developer to join a growing Technology & Innovation team. You'll work on enhancing existing web applications and contribute to new solutions for both B2B and B2C clients. This is an excellent job opportunity to learn from senior developers, gain exposure to modern technologies, and grow into larger responsibilities over time.
£80000 - £90000 per annum
London
Job Title: Enterprise & Technology Risk Manager Location: London (Hybrid - 2 days p/week in office) Type: Full-time Salary: £80,000 - £90,000 + package & bonus
Negotiable
Cambridgeshire
Role: Senior ERP Functional Lead (SCM & PTP) Location: Cambridge (Hybrid Working) Type: Permanent Excellent salary + package on offer
£700 - £750 per day
Berkshire
We're seeking a full stack developer with strong frontend expertise to join an innovation-driven project team. This role will be hands-on, fast-paced, and focused on delivering cutting-edge solutions that incorporate AI and Large Language Models (LLMs). You'll be working on proof-of-concepts, rapid prototypes, and production-ready features in a collaborative, agile environment.
€85000 - €100000 per annum
Bayern (Bavaria)
We're looking for a Senior Data Engineer to join a high-performing Data Engineering team at a fast-growing tech scale-up. You'll help build and scale the pipelines and platforms that power data science, analytics, and product teams, working on systems that process millions of events daily.
£600 - £680 per day
London
Tec Partners are working with a world-leading technology client who are currently looking for an experienced Project Manager, with a background in SAP S/4 HANA and reporting and analytics, to work on an upcoming customer assignment. As a Project Manager, you will take ownership predominantly of reporting and analytics aspects of the programme, a global SAP S/4 HANA upgrade, whilst also getting involved in some business readiness work.
Negotiable
Essex
We are looking for a Database Developer to join a growing technology team, with responsibility for creating and maintaining high-performing database systems. You will play a key role in ensuring that data is well-structured, secure, and accessible, supporting business decision-making and future growth.
£35000 - £45000 per annum
Essex
A growing organisation is seeking a talented .NET Developer to join its collaborative software development team. You will be involved in migrating existing business systems working on both established applications and Greenfield projects.
* Analysing current reporting workflows and proposing practical improvements * Cleansing, mapping, and structuring large data sets * Collaborating with developers to deliver automated reporting solutions * Acting as a change champion for lean, simplified processes
£50000 - £57000 per annum + + benefits
East Sussex
Tec Partners are working with a well-known utilities company, who are looking for an experienced PMO Systems Integration training lead, to come in and create an overall training programme for their capital delivery PMO as they embark on a new system rollout. As a PMO Systems Integration Training Lead, you will make this newly created role your own, owning the overall training strategy, from scope, set-up, rollout and continuous delivery, across their core systems and processes (P6, SAP, PIM ...