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
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. 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
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 continuously (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.
Ansök nu