Postdoc position in probabilistic methods for foundation and world models

SciLifeLab; Uppsala kommun; On-site; SEK 520,000 - 640,000; Full time

Job summary

Uppsala University invites applications for a two-year postdoctoral position in the Scientific Machine Learning group at the Division of Scientific Computing (TDB). The role focuses on uncertainty quantification for large pre-trained models, probabilistic generative and world models, and data-driven scientific discovery.

You will publish and present at international conferences, contribute to open-source software, and may supervise students; teaching may be up to 20%.

Qualifications

• PhD in machine learning, computer science, scientific computing, mathematics, statistics or related field, or foreign degree equivalent; degree must be obtained by the employment decision time.
• Priority for degree within 3 years of deadline; exceptions may apply.
• Documented research in modern deep learning (generative models, Bayesian deep learning, large pre-trained models) and excellent Python skills with PyTorch or JAX.
• Excellent spoken and written English; self-motivation and team-oriented mindset.

Responsibilities

• Research, publish and present results at international conferences.
• Contribute to the group’s open-source software.
• Supervision of students; teaching may be up to 20%.

Skills

Deep learning

Python

PyTorch

JAX

English

Self-motivation

Education

PhD in ML/CS/Math/Statistics

Tools

PyTorch

JAX

Python

Job description

Uppsala University, Disciplinary Domain of Science and Technology, Faculty of Mathematics and Computer Science, Department of Information Technology

Are you interested in working with probabilistic machine learning for the next generation of AI models – uncertainty-aware foundation models, generative models and world models – with the support of competent and friendly colleagues in an international environment? Are you looking for an employer that invests in sustainable employeeship and offers safe, favourable working conditions? We welcome you to apply for a postdoctoral position at Uppsala University.

The Department of Information Technology holds a leading position in both research and education at all levels. We are currently Uppsala University’s third largest department, have around 350 employees, including 120 teachers and 120 PhD students. Approximately 5,000 undergraduate students take one or more courses at the department each year. The department also participates in the Wallenberg AI, Autonomous Systems and Software Program (WASP). You can find more information about us on the Department of Information Technology website.

The position is hosted by the Division of Scientific Computing (TDB), one of the world’s largest research environments in computational science, with large activities in areas such as machine learning, optimization, scientific software development and high-performance computing. The division is an important part of the eSSENCE e-science collaboration and of the Science for Life Laboratory (SciLifeLab) network, a national research infrastructure for life sciences.

The successful candidate will join the Scientific Machine Learning group at TDB and SciLifeLab. The group develops theory, methods and software for data-driven science, with a current focus on uncertainty quantification in large pre-trained models (vision-language models), generative models (flow matching, diffusion), simulation-based inference, and robust and active learning. The group has a wide network of collaborators and strong access to compute through national GPU systems (NAISS, e.g. Berzelius and Arrhenius) and local GPU infrastructure.

Project description

• Uncertainty quantification, calibration and reliability of large pre-trained models
• Probabilistic generative models and world models
• Probabilistic machine learning for scientific discovery
• Don’t see your exact idea listed? We encourage bespoke proposals – outline your own research direction (max 2 pages) within the theme above.

Duties

Research, publication and presentation of results at international conferences, contributions to the group’s open‑source software, and participation in the supervision of students. A limited amount of teaching may be included (max 20%).

Requirements

PhD degree in machine learning, computer science, scientific computing, mathematics, statistics or a related field, or a foreign degree equivalent to a PhD degree in machine learning, computer science, scientific computing, mathematics, statistics or a related field. The degree needs to be obtained by the time of the decision of employment. Priority will be given to applicants who have completed their degree no more than three years before the deadline for applications. Due to special circumstances, the degree may have been obtained earlier. The three-year period can be extended due to circumstances such as sick leave, parental leave, duties in labour unions, etc.

Documented research experience in modern deep learning (e.g. generative models, Bayesian deep learning or large pre-trained models) and excellent programming skills in Python and a modern deep learning framework (e.g., PyTorch or JAX) are required. Excellent skills in spoken and written English are required. The candidate must clearly document a high degree of self‑motivation in the application. Great emphasis will be placed on personal characteristics such as creativity, thoroughness, a structured approach to problem‑solving, and the ability to work both independently and in a team.

Additional Qualifications

Publications at top machine learning or computer vision conferences (NeurIPS, ICML, ICLR, CVPR, AISTATS etc.) are highly meriting. Expertise in Bayesian methods, generative models, multimodal models, world models or simulation‑based inference is meriting, as is experience with large‑scale training on GPU clusters, open‑source software development and applications in the life sciences.

Teaching experience

Teaching experience is considered a merit. This may include teaching, supervision, mentoring, course assistance, providing internal training, or other educational activities, within or outside higher education. In assessing such experience, particular consideration will be given to activities that support students’ learning in computer science, information technology, or closely related subjects. The assessment will take the applicant’s career stage into account, and extensive teaching experience is not expected.

About The Employment

The employment is a temporary position of two years according to central collective agreement. Full time position. Starting date 1 November 2026 or as agreed. Placement: Uppsala

For further information about the position, please contact: Associate Professor Prashant Singh, [e-mail]; Head of Division Elisabeth Larsson, [e-mail].

Are you considering moving to Sweden to work at Uppsala University? Find out more about what it´s like to work and live in Sweden.

Uppsala University is a broad research university with a strong international position. The ultimate goal is to conduct education and research of the highest quality and relevance to make a difference in society. Our most important asset is all of our 7,500 employees and 53,000 students who, with curiosity and commitment, make Uppsala University one of Sweden’s most exciting workplaces.

Read more about our benefits and what it is like to work at Uppsala University

https://uu.se/om-uu/jobba-hos-oss/

The position may be subject to security vetting. If security vetting is conducted, the applicant must pass the vetting process to be eligible for employment.

Please do not send offers of recruitment or advertising services.

Employee organizations: Saco-S – [e-mail], Seko – [e-mail], ST (OFR/S) – [e-mail]