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

Β£55000 - Β£65000 per annum
Cambridgeshire
My client, a high-growth global software business, operating at the forefront of secure remote access technology, is seeking a Senior Software Engineer to join its Portal team. This is an opportunity to work on a business-critical web platform used daily by thousands of customers worldwide and central to the company's commercial and product ecosystem.
Β£75000 - Β£85000 per annum
Remote
Job Title: Senior DevOps Engineer - ArgoCD/GitOps Location: UK - remote Salary: Β£75-85K Type: Permanent
Β£70000 - Β£200000 per annum
Remote
A founding-level Tech Lead / future CTO role at a fully remote, well-funded AI startup building a user-facing "Notion for AI agents." You'll own product and engineering end-to-end, rapidly prototyping and shipping features while also working hands-on with multi-agent systems and AI infrastructure. Small team, long runway, high autonomy, and a clear path to CTO as the company scales.
€70000 - €100000 per annum
Ile de France
A Senior Data Engineer role focused on building the core data infrastructure for a well-funded AI ClimateTech startup. You'll work closely with the CTO to design and scale pipelines in Python, DBT, and PostgreSQL, enabling ML and LLM systems built on complex scientific data. High ownership, early-stage environment, and direct impact on AI products reducing COβ‚‚ emissions.
€80000 - €120000 per annum
Berlin
Join an early-stage team building an applied AI platform for the construction industry, already live with leading enterprises. We're automating the back-office of construction companies by unlocking value from their unstructured data - PDFs, scans, and drawings, saving teams up to 90% of their time.
Β£40000 - Β£50000 per annum
London
An established engineering organisation operating in a highly technical environment is seeking a Software Test Engineer to join its growing team. This role offers the opportunity to work on software that supports complex products and services used in demanding, real-world applications.
Β£27 - Β£31 per hour
Oxfordshire
End User Device Engineer Location: Kidlington Rate: Up to Β£250 per day Umbrella Contract until end of March 2026 with possible extension
Β£70000 - Β£80000 per annum + + benefits
West Sussex
Tec Partners are working with a well-known utilities provider, who are looking for an experienced Software Developer, with React and Python experience, to join their team, focused on protecting the environment and maintaining water quality in the southern region. As a Software Developer, you will be a key technical contributor to one of their flagship digital products, focused on environmental protection. You will work with Reach and TypeScript on the front-end, Python on the back-end...
Tec Partners are working with a world-leading technology client who are currently looking for an experienced Senior Infrastructure Developer, with a strong background in infrastructure automation, Python and VMware Cloud Foundation (VCF) (or similar virtualization technologies). As a Senior Infrastructure Developer, you will automate configuration management and integrate APIs for virtualization platforms using VMware Cloud Foundation (VCF), working on large-scale deployments.
Negotiable
London
Tec Partners are working with a world-leading technology client who are currently looking for an experienced Senior Infrastructure Engineer, with a strong VMware Cloud Foundation (VCF) expertise. As a Senior Infrastructure Engineer, you will deliver virtualization platforms using VMware Cloud Foundation (VCF), working on large-scale deployments.
€50000 - €60000 per annum
Berlin
We are seeking a Java Developer to join a dynamic team delivering cutting-edge solutions within the defence and security sectors. You will play a key role in designing, developing and maintaining the critical backend infrastructure that supports vital services. This is a unique opportunity to work on diverse and challenging projects, contributing to real-world impact within a hybrid and flexible working environment.
€60000 - €70000 per annum
Berlin
We are seeking a Senior Frontend Developer to join a skilled team working on innovative and complex technology solutions within the defence and security sectors. In this role, you will be responsible for designing, developing and maintaining critical frontend applications that support key services. You will work on a variety of challenging projects and contribute to delivering solutions with real-world impact in a hybrid and flexible working environment.