The AI ⋅ Cell Systems Lab

Welcome to the Atanackovic Lab at the University of Alberta! We are a research group focused on developing cutting-edge machine learning methods for understanding and modeling natural (physical) systems from data, with an emphasis on cellular and molecular systems.

Our research spans machine learning, generative modeling, causality, systems biology, single-cell biology, and protein & molecule design. We combine theoretical advances with practical applications to push the boundaries of what’s possible in computational biology and AI.

Interested in joining our lab? Explore our Research areas and learn how to Join Us for opportunities at the intersection of machine learning and biology.

Have questions or want to collaborate? Contact us.

News

Aug 19, 2025 Excited to share some career news! :dna: :sparkles: Starting in 2026, I’ll be joining U of Alberta as an Assistant Professor and Amii as a Research Fellow. Beforehand, I’ll be spending time as a Postdoc Fellow at the Broad Institute.
Jul 31, 2025 I successfully defended my PhD! :mortar_board: Huge thanks to all my mentors, collaborators, and friends who supported me throughout this incredible journey.
Apr 16, 2025 I will be attending ICLR 2025 in Singapore next week – presenting Meta Flow Matching and SuperDiff! :smile:
Feb 20, 2025 I am excited to announce that both Meta Flow Matching (Poster) and SuperDiff (Spotlight) have been accepted to ICLR 2025!
Dec 07, 2024 I will be attending NeurIPS this year in Vancouver! :smile: Reach out if you want to meet and chat!

Selected Publications

  1. SD_examples.gif
    The Superposition of Diffusion Models Using the Itô Density Estimator
    Marta Skreta* , Lazar Atanackovic*, Avishek Joey Bose , Alexander Tong , and Kirill Neklyudov
    International Conference on Learning Representations (ICLR, Spotlight), 2025
    arXiv Code
  2. gif_mfm_letters_train_50.gif
    Meta Flow Matching: Integrating Vector Fields on the Wasserstein Manifold
    Lazar Atanackovic*, Xi Zhang* , Brandon Amos , Mathieu Blanchette , Leo J Lee , Yoshua Bengio , Alexander Tong , and Kirill Neklyudov
    International Conference on Learning Representations (ICLR), 2025
    arXiv Code
  3. gflownet_gen_preview.png
    Investigating Generalization Behaviours of Generative Flow Networks
    Lazar Atanackovic, and Emmanuel Bengio
    Transactions on Machine Learning Research (TMLR), 2025
    arXiv Code
  4. wlfs.png
    A Computational Framework for Solving Wasserstein Lagrangian Flows
    Kirill Neklyudov* , Rob Brekelmans* , Alexander Tong , Lazar Atanackovic, Qiang Liu , and Alireza Makhzani
    International Conference on Machine Learning (ICML), 2024
    arXiv Code
  5. sf2m.png
    Simulation-free Schrödinger Bridges via Score and Flow Matching
    Alexander Tong* , Nikolay Malkin* , Kilian Fatras* , Lazar Atanackovic, Yanlei Zhang , Guillaume Huguet , Guy Wolf , and Yoshua Bengio
    Artificial Intelligence and Statistics (AISTATS), 2024
    arXiv Code
  6. dyngfn.png
    DynGFN: Towards Bayesian Inference of Gene Regulatory Networks with GFlowNets
    Lazar Atanackovic*, Alexander Tong* , Bo Wang , Leo J Lee , Yoshua Bengio , and Jason Hartford
    Advances in Neural Information Processing Systems (NeurIPS), 2023
    arXiv Code

Feature Talks

Software Packages