Machine learning for molecules

Johannes
Schimunek

Research Associate in the Computational and Biological Learning group at the University of Cambridge. I build data-efficient models for small-molecule discovery: few-shot prediction, generative modelling, and chemical reasoning models.

Portrait of Johannes Schimunek leaning against a dark stone wall, looking into the sun
model-card.md · version 2026.09
model
Johannes Schimunek, Research Associate
deployed at
Computational and Biological Learning group, University of Cambridge, with José Miguel Hernández-Lobato
intended use
Few-shot and in-context prediction for molecules, generative models for small molecules, and chemical reasoning models
trained on
Mathematics and physics at the University of Augsburg. Doctorate in machine learning at JKU Linz (2025), supervised by Günter Klambauer and Sepp Hochreiter.
earlier versions
Postdoc at JKU Linz. Maths and physics teacher in Munich. Data scientist in industry.
known for
MHNfs, Bio-xLSTM, MolecularIQ
  1. ICLR 2026

    MolecularIQ: Characterizing Chemical Reasoning Capabilities Through Symbolic Verification on Molecular Graphs

    C. Bartmann*, J. Schimunek*, M. Ielanskyi, P. Seidl, G. Klambauer, S. Luukkonen
  2. ICLR 2025

    Bio-xLSTM: Generative Modeling, Representation and In-Context Learning of Biological and Chemical Sequences

    N. Schmidinger, L. Schneckenreiter, P. Seidl, J. Schimunek, P.-J. Hoedt, J. Brandstetter, A. Mayr, S. Luukkonen, S. Hochreiter, G. Klambauer
  3. J. Chem. Inf. Model. 2025

    MHNfs: Prompting In-Context Bioactivity Predictions for Low-Data Drug Discovery

    J. Schimunek, S. Luukkonen, G. Klambauer
  4. ICLR 2024 workshops

    Autoregressive Activity Prediction for Low-Data Drug Discovery

    J. Schimunek, L. Friedrich, D. Kuhn, G. Klambauer
  5. ICLR 2023

    Context-enriched molecule representations improve few-shot drug discovery

    J. Schimunek, P. Seidl, L. Friedrich, D. Kuhn, F. Rippmann, S. Hochreiter, G. Klambauer
Show all publications (11)Hide full list
  1. ICLR 2026

    MolecularIQ: Characterizing Chemical Reasoning Capabilities Through Symbolic Verification on Molecular Graphs

    C. Bartmann*, J. Schimunek*, M. Ielanskyi, P. Seidl, G. Klambauer, S. Luukkonen
    International Conference on Learning Representations (ICLR)
  2. Nucleic Acids Res. 2026

    Machine learning-driven optimization of specific, compact, and efficient base editors via single-round diversification

    M. Ielanskyi, M. Wang, L. Scott, L. Rieber, S. Merrett, J. Schimunek, A. Mayr, I. McDowell, G. Klambauer, T. Bowen
    Nucleic Acids Research
  3. ESWC 2026

    Structure is the Signal: Graph Encodings and GNNs for Constraint Repair in Collaborative Knowledge Graphs

    M. Vázquez, K. Innerebner, A. Prock, G. Klambauer, E. Lex, J. Schimunek, A. Polleres
    European Semantic Web Conference (ESWC)
  4. ICLR 2025

    Bio-xLSTM: Generative Modeling, Representation and In-Context Learning of Biological and Chemical Sequences

    N. Schmidinger, L. Schneckenreiter, P. Seidl, J. Schimunek, P.-J. Hoedt, J. Brandstetter, A. Mayr, S. Luukkonen, S. Hochreiter, G. Klambauer
    International Conference on Learning Representations (ICLR)
  5. J. Chem. Inf. Model. 2025

    MHNfs: Prompting In-Context Bioactivity Predictions for Low-Data Drug Discovery

    J. Schimunek, S. Luukkonen, G. Klambauer
    Journal of Chemical Information and Modeling
  6. Book chapter 2025

    Few-Shot Learning in Drug Discovery

    J. Schimunek, G. Klambauer
    Book chapter in Applied Artificial Intelligence for Drug Discovery, Springer Nature
  7. Mol. Inform. 2024

    A Community Effort in SARS-CoV-2 Drug Discovery

    J. Schimunek, P. Seidl, K. Elez, et al.
    Molecular Informatics
  8. ICLR 2024 workshops

    Autoregressive Activity Prediction for Low-Data Drug Discovery

    J. Schimunek, L. Friedrich, D. Kuhn, G. Klambauer
    ICLR 2024 workshops: Practical ML for Low Resource Settings, and Data-centric Machine Learning Research (DMLR)
  9. ICLR 2023

    Context-enriched molecule representations improve few-shot drug discovery

    J. Schimunek, P. Seidl, L. Friedrich, D. Kuhn, F. Rippmann, S. Hochreiter, G. Klambauer
    International Conference on Learning Representations (ICLR)
  10. ELLIS ML4Molecules 2021

    A Generalized Framework for Embedding-Based Few-Shot Learning Methods in Drug Discovery

    J. Schimunek, L. Friedrich, D. Kuhn, F. Rippmann, S. Hochreiter, G. Klambauer
    ELLIS Machine Learning for Molecule Discovery Workshop
  11. QSAR 2021

    Comparative Assessment of Interpretability Methods of Deep Activity Models for hERG

    J. Schimunek, L. Friedrich, D. Kuhn, F. Rippmann, S. Hochreiter, G. Klambauer
    International Workshop on (Q)SAR in Environmental and Health Sciences
Google Scholar profile →* equal contribution
  • Invited talk · 2024
    AI for Good Global Summit
  • Invited talk · 2024
    ELLIS Health Symposium on Machine Learning for Drug Discovery
  • Award · 2024
    Best paper, ELLIS ML4Molecules workshop, for Bio-xLSTM
  • Award · 2024
    AI for Good spotlight, ICLR DMLR workshop
  • Award · 2024
    Top Reviewer, NeurIPS
  • Area Chair
    ICLR 2027
  • Reviewer
    ICLR, NeurIPS, ICML, TMLR, Journal of Chemical Information and Modeling, Journal of Medicinal Chemistry
  • Organiser
    ELLIS ML4Molecules workshop, 2024 and 2025. Weekly Drug Discovery Journal Club, since 2020.
  • Teaching
    AI for Life Sciences, Deep Learning and Neural Networks, Hands on AI and Programming in Python at JKU Linz. Supervised more than ten bachelor's and master's theses.

Introducing me at an event? Feel free to use this bio and photo.

Johannes Schimunek is a Research Associate in the Computational and Biological Learning group at the University of Cambridge, working with José Miguel Hernández-Lobato on machine learning for small-molecule discovery when data is scarce: few-shot bioactivity prediction, generative models for chemical and biological sequences, and chemical reasoning in language models. Johannes completed a PhD and a postdoc at the Institute for Machine Learning at JKU Linz with Günter Klambauer and Sepp Hochreiter and developed MHNfs, a few-shot model for bioactivity prediction.