Arbetsbeskrivning

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.

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

• 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. Strong written and spoken English, highly desirable is also fluency in Swedish

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

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

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

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 continuously (deadline 31 jan)

For questions, feel free to reach out to Fazeleh Hoseini, Research Scientist: [email protected]

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