AI Sweden is the national center for applied artificial intelligence, funded jointly by the public and private sectors. Our mission is to accelerate the use of AI to strengthen society, enhance competitiveness, and benefit everyone living in Sweden. A key part of this mission is tackling major research and innovation challenges together with our partners.

We are now looking for a Senior Machine Learning Research Engineer to join our team in Gothenburg. The team works broadly within AI security and advanced machine learning, including privacy-preserving ML, AI Security, computer vision, multimodal learning, and the interplay between algorithms and infrastructure.

About the position

As a Senior ML Research Engineer at AI Sweden, you will contribute to research, experimentation, and development at the intersection of safe, trustworthy, and high-performance AI.

One of our flagship initiatives is LeakPro, an open-source project aimed at detecting and mitigating information leakage in machine learning models. Read more: https://www.ai.se/en/project/leakpro-leakage-profiling-and-risk-oversight-machine-learning-models

Through LeakPro and related efforts, we explore questions such as:

·         How can we systematically measure and reduce unintended information leakage in ML models?

·         How do we build open, reproducible tools for evaluating model vulnerabilities and attack vectors?

·         How can we design and train generative and multimodal models with privacy, integrity, and security at their core?

While privacy-preserving machine learning is an important initial focus, the role can be broader. We seek candidates who want to contribute across several of our current and future research and innovation projects in ML.

Typical tasks may include:

·         Reviewing and implementing cutting-edge ML research

·         Developing and evaluating ML models, attacks, privacy enhanching technologies and defenses.

·         Designing and maintaining open-source tools, benchmarking frameworks, and pipelines

·         Supervising MSc thesis students and collaborating closely with partner organizations

·         Contributing to new project proposals and research/innovation initiatives

We believe that as a person you’re

• Collaborative and team-oriented: You thrive in a collaborative environment and enjoy working with others to achieve shared goals.

• Proactive and solutions-oriented: You take initiative, identify opportunities, and are eager to contribute to solutions.

• Driven by impact: You are motivated by making a positive difference and contributing to meaningful work.

• Curious Learner – passionate about ML research and eager to explore emerging techniques.

• Pragmatic Problem Solver – you navigate complexity with creativity and a practical mindset.

• Clear Communicator – you can articulate complex concepts to diverse audiences.

Technical qualities

Technical qualities

·         Fast Experimenter – adept at prototyping and evaluating new ML methods.

·         ML Systems Thinker – experienced in building reproducible, robust, and scalable ML pipelines.

·         Data Versatile – hands-on experience with at least two data modalities, e.g., images, text, tabular, or time-series.

·         Ethics- and Security-Aware – strong interest in privacy, model safety, and secure deployment practices.

• Strong debugging and refactoring skills in large research codebases.
• Familiarity with differential privacy is a plus.

Requirements

·         PhD in Machine Learning, Computer Science, Data Science, Engineering, or a related field, with a proven applied or research-driven focus OR at least 4+ years of industry experience in machine learning development and research engineering.

• Ability to work independently on complex ML projects and contribute to research-driven development.
• Comfortable reading, understanding, and implementing ideas from research papers.
• Experience working in collaborative codebases and contributing to shared tools or open-source projects.

Technical Expertise

·         Proficiency in Python and deep learning frameworks (PyTorch preferred; TensorFlow also relevant).

·         Experience with generative models (e.g., Transformers/LLMs, diffusion models)

·         Strong understanding of optimization and training dynamics

·         Ability to implement ML research papers from scratch

·         Experience building reproducible ML pipelines and benchmarking frameworks

Language Skills

·         Strong written and spoken English, highly desirable is also fluency in Swedish

·         Experience writing scientific papers, technical documentation, or white papers is an advantage

·         Motivation to contribute to strengthening Sweden’s AI security ecosystem

Why work for AI Sweden?

At AI Sweden, you will join a national initiative supported by over 170 partners across industry, academia, and the public sector. Here, you will:

·         Collaborate with top AI researchers, industry leaders, and innovators

·         Work in an agile, diverse, and inclusive environment

·         Invest in your personal and technical development while building a strong professional network

·         Be part of a mission-driven organization where your ideas can shape the future

An equal and fair working environment

AI Sweden is committed to building a diverse and equitable work environment. We value perspectives from different backgrounds, genders, ages, and experiences.

The position is located at Lindholmen Science Park, Gothenburg, and begins with a two-year contract with the possibility of extension. We are covered by a Swedish collective agreement, offering benefits such as pension, paid leave, wellness allowance, and more.

Employment: Full-time, onsite
Workplace: Kuggen, Lindholmen, Gothenburg

Start date: Early 2026. Applications are evaluated continously (deadline 31 jan)

For questions, feel free to reach out to Fazeleh Hoseini, Research Scientist: fazeleh.hoseini@ai.se

AI Sweden does not accept unsolicited support and kindly ask not to be contacted by any advertisement agents, recruitment agencies or manning companies.