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

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 ...
Β£60000 - Β£75000 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
Β£600 - Β£700 per day
Glasgow City
Tec Partners are working with a world-leading technology client who are currently looking for a number of experienced Azure DevOps Engineers, with strong Azure, development, automation and production deployment experience, to work on an upcoming customer assignment. As an Azure DevOps engineer, you will work on enhancements and be involved in production deployment for a leading financial services organisation.
Job Title: Business Process Improvement Analyst Location: Bury St Edmunds - 3 days in; 2 days remote Job Type: Permanent Salary: Excellent salary & benefits package on offer
Β£40000 - Β£50000 per annum
Norfolk
We're supporting a forward-thinking professional services firm undergoing a major digital transformation. As part of a newly formed team, they're now looking for a mid to senior-level Automation Developer to drive process efficiency through innovative automation solutions. This is a fantastic opportunity to influence the technical direction of a greenfield automation function. You'll work with a mix of technologies including UiPath, Power Automate, APIs, AI/ML, and Document Understanding
US$180000 - US$220000 per annum
New York
A fast moving AI startup is on a mission to help companies understand and control their AI presence. They are hiring a Data Scientist to map every human and AI interaction, helping brands turn AIs into their strongest advocates. You will build predictive models, develop sampling techniques, and optimize brand visibility in AI driven search platforms like ChatGPT and Perplexity. Your work will have a direct impact on how businesses shape their presence in the generative AI era.
US$180000 - US$250000 per annum
New York
A fast moving AI startup is building tools that help companies better understand and manage their AI presence. They are hiring an Analytics Engineer to lay the foundation for product analytics, defining success metrics, mapping the user journey, and uncovering insights that guide product decisions. This role is central to helping the team understand what drives engagement, identify drop off points, and continuously improve the platform's value for users.
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.