Code

Add two ensemble sampling methods#1692

Code

Merged

fehiepsi merged 25 commits intopyro-ppl:masterpyro-ppl/numpyro:masterfrom compmem:ensemblecompmem/numpyro:ensembleCopy head branch name to clipboardJan 26, 2024

+1,055-26

Commits

Commits on Nov 30, 2023

• ensemble sampling draft

  amifalk

  committed
  ef584fc

  Copy full SHA for ef584fc

• rewrite for loop as fori_loop

  amifalk

  committed
  f3b62d2

  Copy full SHA for f3b62d2

Commits on Dec 2, 2023

• added efficiency comment for ESS GaussianMove

  amifalk

  committed
  6fff927

  Copy full SHA for 6fff927

• fix typo

  amifalk

  committed
  dc582e5

  Copy full SHA for dc582e5

Commits on Dec 24, 2023

• Merge branch 'master' into ensemble

  amifalk

  committed
  fd3702e

  Copy full SHA for fd3702e

• fixed ravel for mixed dtype

  amifalk

  committed
  49e4cdb

  Copy full SHA for 49e4cdb

Commits on Jan 21, 2024

• Merge branch 'pyro-ppl:master' into ensemble

  amifalk

  authored
  2f6324a

  Copy full SHA for 2f6324a

• add defaults

  amifalk

  committed
  bc7c792

  Copy full SHA for bc7c792

• add support for potential_fn

  amifalk

  committed
  1ebff6c

  Copy full SHA for 1ebff6c

• AIES tests, warnings for AIES

  amifalk

  committed
  3625d75

  Copy full SHA for 3625d75

• AIES input validation

  amifalk

  committed
  cc501a4

  Copy full SHA for cc501a4

• better docs, more input validation

  amifalk

  committed
  73ca58d

  Copy full SHA for 73ca58d

Commits on Jan 22, 2024

• ESS passing test cases

  amifalk

  committed
  e368781

  Copy full SHA for e368781

• add tests for other files

  amifalk

  committed
  ff38013

  Copy full SHA for ff38013

• linting

  amifalk

  committed
  78945bf

  Copy full SHA for 78945bf

• refactor ensemble_util

  amifalk

  committed
  b4a9695

  Copy full SHA for b4a9695

• make test result less close to margin in CI, swap deprecated function

  amifalk

  committed
  d865eac

  Copy full SHA for d865eac

Commits on Jan 23, 2024

• rename get_nondiagonal_indices, fix batch_ravel_pytree

  amifalk

  committed
  d2b5a09

  Copy full SHA for d2b5a09

• print ensemble kernel diagnostics, smoke test parallel arg

  amifalk

  committed
  b20829d

  Copy full SHA for b20829d

• fix docstring build

  amifalk

  committed
  84bf11c

  Copy full SHA for 84bf11c

• documentation

  amifalk

  committed
  33133e4

  Copy full SHA for 33133e4

• skip slow CI tests, unnest test if statements

  amifalk

  committed
  2c325a1

  Copy full SHA for 2c325a1

• fix doctest

  amifalk

  committed
  e35d34a

  Copy full SHA for e35d34a

Commits on Jan 24, 2024

• doc rewrite

  amifalk

  committed
  550b023

  Copy full SHA for 550b023

Commits on Jan 25, 2024

• fix distribution test

  amifalk

  committed
  96e9483

  Copy full SHA for 96e9483