math.CTAug 14, 2026

Sample-Smooth Spaces: A Convenient Category for Differentiable Probabilistic Programming

Authors: Patrick Forré

Organizations: AI4Science Lab Korteweg-de Vries Institute for Mathematics University of Amsterdam

Abstract

We introduce the category SSS\mathbf{SSS} of sample-smooth spaces over a mixed site. The test objects are the products Ωn:=Rn×ΩΩ_n := \mathbb{R}^n \times Ω of a Cartesian space with the universal Hilbert cube ΩΩ carrying all universally measurable sets, and a space is a set with a family of admissible plots Ωn→XΩ_n \to \mathcal{X} closed under precomposition. Smoothness and measurability are then not two structures glued along an axiom, but one structure over one site. The site has finite non-empty products, because ΩΩ absorbs its own square; its Karoubi envelope contains every Rn\mathbb{R}^n; and it has mixed morphisms ω↦(W(ω),Φ(ω))ω\mapsto (W(ω),Φ(ω)), which turn measurability of a smooth family from an axiom into a consequence. SSS\mathbf{SSS} is a concrete quasitopos: complete, cocomplete, cartesian closed and locally cartesian closed, with a classifier for embeddings. Morphisms of Cartesian spaces are exactly the C∞C^\infty maps and manifolds embed full and faithfully, both without Boman's theorem. Every object has tangent and cotangent spaces, every morphism a differential. The modalities sit in an adjoint string Π⊣♭⊣♮⊣♯⊣ΛΠ\dashv \flat \dashv \natural \dashv \sharp \dashv Λ, making SSS\mathbf{SSS} cohesive over quasi-universal spaces. The point is the probability monad. Defining the plots of P(X)\mathsf{P}(\mathcal{X}) as push-forwards of X\mathcal{X}-plots at every test object, P\mathsf{P} is an unconditional strong commutative affine monad on all of SSS\mathbf{SSS} -- functor, unit, product of kernels, multiplication and the monad laws are each one line of seed splitting -- and its Kleisli category, of differentiable simulators, is a Markov category. The reparametrisation trick holds by construction: every Kleisli morphism is plot-wise a sampler, stably under composition. A reflection theorem locates the whole gain in a single plot family.

Figures & tables

Explore similar work

CardsList
  1. NeSyCat Torch: A Differentiable Tensor Implementation of Categorical Semantics for Neurosymbolic Learning

    Jun 17, 2026Daniel Romero Schellhorn, Till Mossakowski, Björn GehrkeNeuro-Symbolic ReasoningNeurosymbolic Learning

  2. The Normalized Maximum Likelihood for Regular Non-Smooth Models: Measure-Theoretic Foundations and Geometric Sampling

    May 23, 2026Trenton Lau, Gary P. T. ChoiIndependent Metropolis-HastingsMarkov Chain Monte Carlo

  3. Riemannian Denoising Diffusion Probabilistic Models

    May 7, 2025Zichen Liu, Wei Zhang, Christof Schütte +1Generative ModelingScore-Based Generative Modeling