Rafal Urbaniak

Causal inference and probabilistic machine learning at scale

Rafal Urbaniak

I am a research scientist at the Basis Research Institute (New York/Cambridge, MA), a nonprofit AI research-and-engineering lab building a general-purpose reasoning engine. I work on causal inference and probabilistic machine learning, and on making them run at scale, from formal method to production system. I am a core contributor to ChiRho, Basis’s open-source causal probabilistic programming language (built on Pyro/PyTorch). Among other things, I lead the work on its Explainable module, the probabilistic causal impact attribution machinery.

My research builds causal Bayesian models with deep machine-learned components for attribution and counterfactual reasoning under uncertainty in complex, partly observed dynamical systems, with a recent emphasis on spatio-temporal problems. One recent paper gives a counterfactual semantics for hybrid dynamical systems (NeurIPS 2025).

I hold a PhD in Logic and Philosophy of Mathematics from the University of Calgary, and have held positions at the Research Foundation – Flanders, Trinity College Dublin, the University of Bristol, and the University of Gdańsk.

Currently working on

Open source

Most of my engineering work happens in private repositories. The chart below summarizes my GitHub activity by year: pull requests authored and merged (top), reviews and issues (bottom), across public and private repositories.

Pull requests 0 50 100 150 PRs authored PRs merged Reviews & issues 0 70 140 210 2020 2021 2022 2023 2024 2025 2026 Reviews Issues

GitHub activity per year: pull requests authored and merged (top); reviews and issues (bottom). Public and private repositories combined; 2026 is year-to-date.

Elsewhere