Search results

eca.py: ensemble_execute_fn closes over sharded JAX arrays in shard_map — NotImplementedError on N>1 devices

Status: Open.

#932 In blackjax-devs/blackjax;

· junpenglao opened on May 30, 2026

Feature request: extend the standard warmup & inference utilities to be more vmap-friendly (multi-chain)

Status: Open.

#927 In blackjax-devs/blackjax;

· junpenglao opened on May 27, 2026

MAKLA-BCSS2 sampler

Status: Open.

#913 In blackjax-devs/blackjax;

· paulindani opened on May 13, 2026

diagnostics: add rank-normalized + folded R-hat (Vehtari et al. 2021)

Status: Open.

#912 In blackjax-devs/blackjax;

· junpenglao opened on May 12, 2026

Move logdensity_fn to build_kernel and pass logdensity_params at call time

Status: Open.

#881 In blackjax-devs/blackjax;

· junpenglao opened on Apr 15, 2026

Implement Slice Sampler Reparameterization Gradients

Status: Open.

#845 In blackjax-devs/blackjax;

· ischeinfeld opened on Mar 31, 2026

Add VSBC diagnostic for evaluating variational inference calibration

enhancementNew feature or requestNew feature or request

Status: Open.

#840 In blackjax-devs/blackjax;

· junpenglao opened on Mar 26, 2026

Add Kernelized Stein Discrepancy (KSD) as a convergence diagnostic

enhancementNew feature or requestNew feature or request

Status: Open.

#839 In blackjax-devs/blackjax;

· junpenglao opened on Mar 26, 2026

Add gradient-based HMC hyperparameter adaptation via Stein discrepancy (Campbell et al. 2021)

samplerIssue related to samplersIssue related to samplers

Status: Open.

#838 In blackjax-devs/blackjax;

· junpenglao opened on Mar 26, 2026

Add discrete bouncy particle sampler (Sherlock & Thiery)

samplerIssue related to samplersIssue related to samplers

Status: Open.

#837 In blackjax-devs/blackjax;

· junpenglao opened on Mar 26, 2026

Add AMAGOLD: amortized Metropolis-adjusted stochastic gradient MCMC

samplerIssue related to samplersIssue related to samplers

Status: Open.

#836 In blackjax-devs/blackjax;

· junpenglao opened on Mar 26, 2026

Add discrete Langevin sampler (DLP / DMALA) for discrete distributions

samplerIssue related to samplersIssue related to samplers

Status: Open.

#835 In blackjax-devs/blackjax;

· junpenglao opened on Mar 26, 2026