Postdoc Position in Post-Training of Diffusion Language Models (Prof. I. Bogunovic)

Academic Positions; Schweiz; On-site; CHF 90,000 - 120,000; Full time

Benefits offered by this job

Travel to international conferences

Mentoring PhD students

Excellent salary and benefits

Job summary

The Rhine AI group of Prof. I.

Bogunovic within the Department of Mathematics and Computer Science at the University of Basel invites applications for a Postdoc position funded by the SNSF project Foundations of Post-Training for Reasoning in Diffusion Large Language Models. Starting January 2027 or later, you will advance post-training methods for diffusion-based LLMs and develop benchmarks for reasoning across domains.

Qualifications

• PhD in machine learning or related field; strong publication record in leading ML venues.
• Strong mathematical foundation and coding/implementation skills.
• Experience with LLMs, reinforcement learning, or generative modeling is a plus.
• Excellent communication skills in English (spoken and written).

Responsibilities

• Conduct research in machine learning and AI.
• Take a leading role in project directions and mentor PhD students.
• Publish papers and present results at major conferences (ICML, NeurIPS, ICLR).
• Collaborate with national and international AI/ML teams.

Skills

Machine learning

Reinforcement learning

Generative modeling

Coding & implementation

English fluency

Education

PhD in machine learning or related field

Tools

Python

PyTorch

Job description

Organisation/Company University of Basel Research Field Computer science » Other Researcher Profile Recognised Researcher (R2) Application Deadline 2 Dec 2026 - 22:59 (UTC) Country Switzerland Type of Contract Temporary Job Status Full-time Is the job funded through the EU Research Framework Programme? Not funded by a EU programme Is the Job related to staff position within a Research Infrastructure? No

Offer Description

The research group Rhine AI of Prof. I. Bogunovic at the Department of Mathematics and Computer Science at the University of Basel is inviting applications for a Postdoc position funded by the Swiss National Science Foundation (SNSF) project "Foundations of Post-Training for Reasoning in Diffusion Large Language Models." The position focuses on post-training large diffusion-based language models (dLLMs), an emerging alternative to standard autoregressive LLMs, to improve their reasoning capabilities. The successful candidate will develop reinforcement learning methods for post-training large dLLMs, explore reasoning-aware generation strategies such as adaptive unmasking and credit assignment, and help build benchmarks for evaluating reasoning across domains, including mathematical problem solving. The ideal candidate has a strong background in machine learning and is excited about large-scale experimentation with modern AI systems, supported by principled analysis where it helps. This work will push the boundaries of what modern AI systems can reliably reason about.

The position is available starting January 2027 or later.

• Conduct research in machine learning and artificial intelligence
• Take a leading role in the project's research directions and help mentor PhD students
• Write research papers and articles, and present results at leading international conferences. We publish in venues such as ICML, NeurIPS, and ICLR
• Interact with national and international teams of AI and machine learning experts

Applicants must hold a PhD in machine learning or a related field, with a strong publication record at leading machine learning venues. Candidates should have an exceptional academic record and a solid mathematical foundation. They are also expected to have strong coding and implementation skills, with the ability to run large-scale experiments. Experience with LLMs, reinforcement learning, or generative modeling is a plus.

Excellent communication skills and fluency in English (spoken and written) are required.

We also seek to increase the number of women and other groups in areas where they are underrepresented and therefore explicitly encourage such groups to apply.

The position will be based in Basel, Switzerland. In addition, we offer:

• Close collaboration and mentoring, working with and access to national (EPFL and ETH among others) and international collaborators
• A diverse range of tasks in a dynamic environment
• The possibility of traveling to international conferences to meet distinguished scientists
• Excellent salary, benefits, and working conditions