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

US$150000 - US$200000 per annum
New York
Data Scientist We're looking for a Data Scientist to help establish the quantitative foundation of a cutting-edge trust and validation framework for autonomous systems. In this role, you'll design rigorous statistical methodologies to evaluate system performance, develop confidence and reliability metrics, and support high-scale deployment with robust measurement systems.
€80000 - €120000 per annum + 120000
Berlin
Founding AI Engineer - Berlin (German fluency required) €100k Β± 20% | equity | Full-time, on-site We're building the AI automation layer for the construction industry - live at major enterprises and already saving teams up to 90% of back-office time. Join us in Berlin as our Founding AI Engineer to shape and scale high-impact, real-world GenAI use cases from day one.
US$180000 - US$250000 per annum
New York
Data Engineer We're looking for an experienced Data Engineer to take ownership of our data platform and help scale it to meet growing demands. You'll be responsible for maintaining and optimizing data pipelines, designing new ingestion processes, and supporting the data needs of various teams including product, data science, and go-to-market.
Β£40000 - Β£60000 per annum
Norfolk
Are you a skilled Full Stack Developer with a passion for clean code, smart systems, and meaningful tech? We're working with an innovative environmental monitoring business that's on a mission to reshape how we interact with scientific data - and they need their first dedicated developer to help lead that charge. This is a unique opportunity to join a close-knit team as they rebuild their data-driven platform from the ground up.
Β£60000 - Β£70000 per annum
Cheshire
Requirements: * 5+ years of front-end experience * Expert in React and TypeScript * Hands-on AWS integration experience * Strong understanding of APIs (REST/GraphQL) * CI/CD familiarity and clean coding practices
Β£35000 - Β£40000 per annum
London
Company TEC Partners are representing one of the UK's fastest growing competition companies. They have amassed a large social media following having given out over Β£100 million in luxury prizes, including cars, tech and even houses! With a 4.5 Trustpilot rating and a loyal and continually expanding customer base, they have positioned themselves as one of the hottest companies to work for.
Β£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.
US$180000 - US$250000 per annum + Equity
California
I'm working with a YC-backed startup building an IDE for creating robust AI agents in natural language, think Notion for LLM systems. They need an Applied AI Engineer to design multi-step agents, RAG pipelines, memory, and evals. Full-time, in person in SF. Game-changing work at the edge of AI reliability.
Β£125000 - Β£150000 per annum + + double OTE
New York
Series A devtools startup ($60M raised) is hiring Enterprise AEs to sell a high-ACV, AI-powered platform that helps engineers rapidly build internal apps-selling into technical stakeholders (CTOs, VPs Eng, etc).
US$120000 - US$180000 per annum + Equity
California
Join a Y Combinator-backed AI startup as a Founding ML Engineer to build tools that test and monitor LLM-powered voice agents. You'll develop core infrastructure, real-time evaluation pipelines, and user-facing features. Ideal for engineers with strong Python, ML, and LLM experience who want to shape critical AI reliability tools.
US$125000 - US$150000 per annum + + double OTE
Texas
We're hiring a Founding Account Executive to lead GTM at a stealth AI startup making internal data conversational through LLMs. Work directly with a proven, multi-time founder to shape the sales motion from day one.
Β£350 - Β£450 per day
West Midlands
Job Title: Network Support Engineer Location: Birmingham - Hybrid Job Type: Contract (Inside IR35) - 9-12 months Rate: Β£350-450 p/day