Senior Machine Learning Research Engineer
Description
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 -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, integrity, and security at their core?
While -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, enhanching technologiesand defenses.
·Deg 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.
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
·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, model safety, and secure deployment practices.
• Strong debugging and refactoring skills in large research codebases.• Familiarity with differential is a plus.
Requirements
·PhD in Machine Learning, Computer Science, Data Science, Engineering, or a related field, with a proven applied or research-driven focusORat least4+ years of industry experiencein 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, onsiteWorkplace: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:inquiries@consultant.dev
AI Sweden does not accept unsolicited support and kindly ask XXXX XXXX be contacted by any advertisement agents, recruitment agencies or manning companies.