Add ensemble sampling methods to BlackJAX - #797

Open

williamjameshandley wants to merge 18 commits into

blackjax-devs:mainfrom

handley-lab:ensemble

Commits on Jul 8, 2025

Implement emcee stretch move ensemble sampler

williamjameshandley
and
claude
committed
c1b3d8f

Commits on Oct 15, 2025

Remove PLAN.md and REVIEW.md
williamjameshandley
committed
526e5f5
Fix ensemble sampler correctness issues and add emcee compatibility features

williamjameshandley
and
claude
committed
923f6ce
Fix remaining PyTree bugs and implement nsplits > 2 support

williamjameshandley
and
claude
committed
1032cf9
Refactor ensemble sampler for BlackJAX consistency

williamjameshandley
and
claude
committed
7b57761
Merge stretch.py into ensemble.py and rename to stretch.py

williamjameshandley
and
claude
committed
d5b2745
Rename stretch.py back to ensemble.py

williamjameshandley
and
claude
committed
d54b803
Remove stretch alias and expose ensemble as public API

williamjameshandley
and
claude
committed
4503fed
Add Ensemble Slice Sampling (ESS) implementation

williamjameshandley
and
claude
committed
86b10f7
Refine ensemble slice implementation and follow house style

williamjameshandley
and
claude
committed
925f468
Extract shared ensemble utilities to reduce code duplication

williamjameshandley
and
claude
committed
7921cea
Further reduce ensemble code duplication

williamjameshandley
and
claude
committed
fc977e3
Remove unnecessary unshuffle_1d helper

williamjameshandley
and
claude
committed
4db5872
Use vmapped_logdensity in stretch._update_half

williamjameshandley
and
claude
committed
8c247e8

Commits on Oct 16, 2025

Merge branch 'main' into ensemble
junpenglao
authored
d299817

Commits on Oct 29, 2025

Merge branch 'main' into ensemble
junpenglao
authored
6f8054d

Commits on Nov 15, 2025

Merge branch 'main' into ensemble
junpenglao
authored
f0b9e09

Commits on Nov 18, 2025

Merge branch 'main' into ensemble
junpenglao
authored
18b0dd9