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

£35000 - £40000 per annum
Norfolk
This is a pivotal leadership role responsible for service performance, client satisfaction, operational governance, and continual service improvement. You will act as the primary operational contact for key accounts, ensuring SLA/XLA adherence, managing escalations, and driving consistent, high-quality service delivery.
£30000 - £40000 per annum
Norfolk
I'm proud to be recruiting for a well-established software company at the forefront of retail technology, as they look to welcome a Software Developer to their growing team. This hybrid role requires one day a week onsite and offers the chance to work on a cutting-edge, industry-leading software platform used by major retail clients across the UK and beyond. You'll be a well-rounded developer with 2+ years' commercial experience, confident across both front- and back-end technologies.
10% pension
Suffolk
I'm working with a UK-based infrastructure organisation at the forefront of innovation and sustainability, seeking a Senior Software Developer. This is an exciting opportunity to play a key role in designing, developing, and supporting enterprise-level applications while contributing to a collaborative and forward-thinking technical environment. As a Senior Software Developer, you'll work closely with stakeholders, ensuring solutions are fit for purpose, customer-focused, and aligned to business
£50000 - £65000 per annum
Hertfordshire
A growing, innovation-led biotechnology company based in the Cambridge area is seeking a Senior Software Engineer to support the development of advanced instrumentation platforms used in cutting-edge scientific research. This is a fantastic opportunity to work on complex systems that combine software, hardware, and data analysis to deliver real-world impact in life sciences. You'll play a key role in designing and developing high-quality application and control software across Windows and Linux
£70000 - £80000 per annum
Norfolk
This is a high-impact, cross-functional role focused on improving how work actually gets done. You'll work across teams, systems and processes to reduce friction, improve clarity, and build ways of working that are scalable, reliable and easy to use.
£500 - £525 per day
Remote
Tec Partners are working with a world-leading technology client who are currently looking for an experienced Technical Business Analyst, to work on an exciting customer assignment. As a Technical Business Analyst, you will support on a GCP and cloud security adoption focussed transformation programme.
GCP Cloud Security Engineer / GCP Security Expert Contract, 6 months initially (very likely extensions) Inside IR35 Remote (UK) £550-600 per day
£425 - £475 per day
Suffolk
Advanced experience with C# and .NET (Framework and Core) Frontend development using TypeScript, HTML, CSS, and either React or Vue Proven track record with AWS, ideally including serverless architectures (e.g., Lambda) Experience with CI/CD pipelines and Infrastructure as Code (Terraform or CDK) Strong grounding in clean architecture, SOLID principles, and code maintainability Knowledge of API integration and secure development practices (OWASP)
£500 - £510 per day
London
Tec Partners are working with a world-leading technology client who are currently looking for an experienced Network Engineer, to work on an exciting customer assignment. As a Network Engineer, you will join an internal infrastructure operations function, acting as the support and escalation point for a complex, multi-site campus network estate. You will play a key role in ensuring operational reliability and availability of core infrastructure.
£450 - £550 per day
Devon
We are working with a leading UK organisation to find an experienced Product Manager for a key role within their Platform Technology function. Our client is undergoing a significant technology modernisation programme and needs a skilled product professional to help shape and improve the internal platforms their developers, scientists and service teams depend on every day.
£30000 - £45000 per annum
Essex
We're working with an award-winning Financial SaaS business - one of the UK's biggest facilitators of commercial finance. They're growing their engineering team and are looking for a talented .NET Developer with a frontend lean to help shape the next generation of their products.
€80000 - €120000 per annum
Remote
Building an AI that can directly edit live products, understand design systems, and generate production-ready UI. Not from scratch, not via repo access. Real products, real context. You'll own the agentic intelligence that makes this possible.