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

£50000 - £55000 per annum
Cheshire
A fast-growing, technology-led organisation is seeking a Full Stack Java Developer to join its collaborative engineering team. This role is well-suited to someone who's not just looking for a coding job, but who wants to get stuck in - someone with a clear desire to learn, take on ownership, and make a real contribution to the wider business and its platforms.
£45000 - £50000 per annum
Cambridgeshire
Role: Software Test Engineer Location: Hybrid (Cambridge / London) Salary: Up to £50,000 We're working with an established technology business who have been at the forefront of remote access and connectivity solutions for over two decades. Their software is trusted by millions of users globally - from leading names in technology and engineering through to organisations operating in healthcare, media, and industry.
£84000 - £94000 per annum
London
Position: Security Lead / Security Architect Type: Permanent Location: London 2 days p/week; 3 days remote Salary: £84K - £94K DOE
€500 - €600 per day
Kerry
Tec Partners are working with a world-leading technology client who are currently looking for an experienced Infrastructure Project Manager with a technical backbone to run the IT/OT Infrastructure Workstream on a customer assignment within a live construction environment. As an Infrastructure Project Manager, you will own the infrastructure workstream schedule and dependencies across M&E, security, networks, server/compute, storage and site readiness.
£200 - £250 per day
Norfolk
Senior IT Support Engineer - 6-Month Contract On-site | Norwich £200-£250 per day (via Umbrella)
£70000 - £100000 per annum
Cheshire
Key Skills & Experience: 5+ years of frontend development experience Strong commercial experience with React and TypeScript Familiarity with Vite, Next.js, or other modern build tools Comfortable with component libraries and design systems Proficient in testing frameworks (e.g. Jest, React Testing Library) Good understanding of performance optimisation and accessibility best practices
£40000 - £60000 per annum
Hertfordshire
What we're looking for: * 3+ years' experience in software engineering * Strong background with C# and .NET * Familiarity with Linux, Python, and Azure DevOps * Experience in both start-up/scale-up and larger organisations * Excellent communication skills * Bonus: experience in robotics or hardware-based products
£45000 - £55000 per annum
Essex
We're looking for a skilled Software Developer to join a growing technology team at a fast-moving online retail platform. You'll be involved in building new features, improving system performance, and shaping the customer experience on both web and mobile.
£600 - £700 per day
West Yorkshire
We're looking for a Lead Java Developer to join a major national digital transformation programme. The successful candidate will help build and improve high-traffic citizen-facing services used by millions across the UK. You'll lead or mentor a team of Java developers, deliver robust, scalable solutions, and contribute to modernising a complex digital estate. This is a hands-on technical leadership role for someone who thrives in collaborative, Agile environments.
£60000 - £100000 per annum
Cambridgeshire
This is a unique opportunity to join a business where technology underpins everything they do. You'll lead by example, championing best-practice design patterns, robust software architecture, and agile delivery. As the company prepares for significant growth, you'll help re-architect and harden systems to handle 10x-100x scale, ensuring the platform is built to last.
Negotiable
London
Position: Technical Business Analyst (Data) Type: Contract 3 months initially w/ likely extensions IR35: Inside Scope Working Arrangements: 3 days p/week in London office, 2 days remote
Negotiable
London
Position: Technical Architect (Data) Type: Contract 3 months initially w/ likely extensions IR35: Inside Scope Working Arrangements: 3 days p/week in London office, 2 days remote