Add ensemble sampling methods to BlackJAX#797

Open

williamjameshandley wants to merge 18 commits intoblackjax-devs:mainfrom

handley-lab:ensemble

Commits on Jul 8, 2025

Implement emcee stretch move ensemble sampler

williamjameshandley
and
claude
committed

Commits on Oct 15, 2025

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

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

williamjameshandley
and
claude
committed
Refactor ensemble sampler for BlackJAX consistency

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

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

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

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

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

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

williamjameshandley
and
claude
committed
Further reduce ensemble code duplication

williamjameshandley
and
claude
committed
Remove unnecessary unshuffle_1d helper

williamjameshandley
and
claude
committed
Use vmapped_logdensity in stretch._update_half

williamjameshandley
and
claude
committed

Commits on Oct 16, 2025

Merge branch 'main' into ensemble
junpenglao
authored

Commits on Oct 29, 2025

Merge branch 'main' into ensemble
junpenglao
authored

Commits on Nov 15, 2025

Merge branch 'main' into ensemble
junpenglao
authored

Commits on Nov 18, 2025

Merge branch 'main' into ensemble
junpenglao
authored