Make `Delta.log_prob` jit-able on metal and raise explicit error for `Delta` sampling site during initialization.#1950

Merged

fehiepsi merged 2 commits intopyro-ppl:masterfrom tillahoffmann:delta-density

Jan 17, 2025

+29-1

Commits on Jan 16, 2025

Use jnp.where for Delta.log_prob (cf. jax-ml/jax#25935 ).
tillahoffmann
committed
Raise explicit error for unobserved Delta sample sites during intialization.
tillahoffmann
committed