Code

Add MCLMC sampler#586

Code

Merged

junpenglao merged 91 commits intoblackjax-devs:mainblackjax-devs/blackjax:mainfrom reubenharry:refactorreubenharry/blackjax:refactorCopy head branch name to clipboardDec 5, 2023

+567-1

Commits

Commits on Nov 10, 2023

• initial draft of mclmc

  reubenharry

  committed
  fbe7f75

  Copy full SHA for fbe7f75

Commits on Nov 11, 2023

• refactor

  reubenharry

  committed
  3a23242

  Copy full SHA for 3a23242

• wip

  reubenharry

  committed
  86b3a90

  Copy full SHA for 86b3a90

• wip

  reubenharry

  committed
  e82550f

  Copy full SHA for e82550f

• wip

  reubenharry

  committed
  f0e1bec

  Copy full SHA for f0e1bec

• wip

  reubenharry

  committed
  4d7dc57

  Copy full SHA for 4d7dc57

• wip

  reubenharry

  committed
  82b8466

  Copy full SHA for 82b8466

• fix pre-commit

  reubenharry

  committed
  a4d403b

  Copy full SHA for a4d403b

• remove dim from class

  reubenharry

  committed
  a67ecb7

  Copy full SHA for a67ecb7

• add docstrings

  reubenharry

  committed
  3dd4f74

  Copy full SHA for 3dd4f74

Commits on Nov 13, 2023

• add mclmc to init

  reubenharry

  committed
  5d8061d

  Copy full SHA for 5d8061d

• Merge branch 'main' of https://github.com/blackjax-devs/blackjax

  reubenharry

  committed
  5428f2c

  Copy full SHA for 5428f2c

• Merge branch 'main' into refactor

  reubenharry

  committed
  59ecc8a

  Copy full SHA for 59ecc8a

• move minimal_norm to integrators

  reubenharry

  committed
  2bf639e

  Copy full SHA for 2bf639e

• move update pos and momentum

  reubenharry

  committed
  172fee0

  Copy full SHA for 172fee0

• remove params

  reubenharry

  committed
  b710e62

  Copy full SHA for b710e62

Commits on Nov 14, 2023

• Infer the shape from inverse_mass_matrix outside the function step

  reubenharry

  committed
  3cc52fd

  Copy full SHA for 3cc52fd

• use tree_map

  reubenharry

  committed
  57d5c3b

  Copy full SHA for 57d5c3b

Commits on Nov 15, 2023

• integration now aligned with mclmc repo

  reubenharry

  committed
  7e70d78

  Copy full SHA for 7e70d78

• dE and logdensity align too (fixed sign error)

  reubenharry

  committed
  1343463

  Copy full SHA for 1343463

• make L and step size arguments to kernel

  reubenharry

  committed
  e53a877

  Copy full SHA for e53a877

• rough draft of tuning: works

  reubenharry

  committed
  05517b6

  Copy full SHA for 05517b6

• remove inv mass matrix

  reubenharry

  committed
  d84a23d

  Copy full SHA for d84a23d

• almost correct

  reubenharry

  committed
  de1e5cf

  Copy full SHA for de1e5cf

Commits on Nov 16, 2023

• almost correct

  reubenharry

  committed
  263ab3a

  Copy full SHA for 263ab3a

• move tuning to adaptation

  reubenharry

  committed
  777213d

  Copy full SHA for 777213d

• tuning works in this commit

  reubenharry

  committed
  e75274a

  Copy full SHA for e75274a

• clean up 1

  reubenharry

  committed
  8a89f13

  Copy full SHA for 8a89f13

• remove sigma from tuning

  reubenharry

  committed
  49b3bec

  Copy full SHA for 49b3bec

• wip

  reubenharry

  committed
  81999f9

  Copy full SHA for 81999f9

Commits on Nov 17, 2023

• fix linting

  reubenharry

  committed
  8ab01f2

  Copy full SHA for 8ab01f2

• rename T and V

  reubenharry

  committed
  6266bc4

  Copy full SHA for 6266bc4

• uniformity wip

  reubenharry

  committed
  ca984e7

  Copy full SHA for ca984e7

• make uniform implementation of integrators

  reubenharry

  committed
  59ffb21

  Copy full SHA for 59ffb21

Commits on Nov 18, 2023

• make uniform implementation of integrators

  reubenharry

  committed
  8f9214f

  Copy full SHA for 8f9214f

• fix minimal norm integrator

  reubenharry

  committed
  b2e3b8e

  Copy full SHA for b2e3b8e

• add warning to tune3

  reubenharry

  committed
  2fb2293

  Copy full SHA for 2fb2293

Commits on Nov 19, 2023

• Refactor integrators.py to make it more general.

  Show description for 59e4424
  

  junpenglao

  and

  reubenharry

  committed
  59e4424

  Copy full SHA for 59e4424

• temp: explore

  reubenharry

  committed
  6684413

  Copy full SHA for 6684413

Commits on Nov 20, 2023

• Refactor to use integrator generation functions

  junpenglao

  committed
  4284092

  Copy full SHA for 4284092

• Additional refactoring

  Show description for 4a514dd
  

  junpenglao

  and

  reubenharry

  committed
  4a514dd

  Copy full SHA for 4a514dd

Commits on Nov 21, 2023

• Minor clean up.

  junpenglao

  committed
  ef1f62d

  Copy full SHA for ef1f62d

• Use standard JAX ops

  junpenglao

  committed
  af43521

  Copy full SHA for af43521

Commits on Nov 23, 2023

• new integrator

  reubenharry

  committed
  0dd419d

  Copy full SHA for 0dd419d

• add references

  reubenharry

  committed
  0c8330e

  Copy full SHA for 0c8330e

• merge

  reubenharry

  committed
  e6fa2bb

  Copy full SHA for e6fa2bb

Commits on Nov 24, 2023

• flake

  reubenharry

  committed
  40fc61c

  Copy full SHA for 40fc61c

• temporarily add 'explore'

  reubenharry

  committed
  6ea5320

  Copy full SHA for 6ea5320

Commits on Nov 25, 2023

• temporarily add 'explore'

  reubenharry

  committed
  c83dc1a

  Copy full SHA for c83dc1a

Commits on Nov 26, 2023

• Adding a test for energy preservation.

  Show description for c8b43be
  

  junpenglao

  and

  reubenharry

  committed
  c8b43be

  Copy full SHA for c8b43be

• fix formatting

  junpenglao

  committed
  8894248

  Copy full SHA for 8894248

• wip: tests

  reubenharry

  committed
  9865145

  Copy full SHA for 9865145

• Merge branch 'integrator_refactor' into refactor

  reubenharry

  committed
  68464bc

  Copy full SHA for 68464bc

• use pytrees for partially_refresh_momentum, and add test

  reubenharry

  committed
  0c61412

  Copy full SHA for 0c61412

Commits on Nov 27, 2023

• Merge branch 'main' into refactor

  junpenglao

  authored
  a66af60

  Copy full SHA for a66af60

• update docstring

  reubenharry

  committed
  be07631

  Copy full SHA for be07631

• resolve conflict

  reubenharry

  committed
  71d934b

  Copy full SHA for 71d934b

• remove 'explore'

  reubenharry

  committed
  a170d0b

  Copy full SHA for a170d0b

• fix pre-commit

  reubenharry

  committed
  8cfb75f

  Copy full SHA for 8cfb75f

Commits on Nov 29, 2023

• adding randomized MCHMC

  JakobRobnik

  committed
  b42e77e

  Copy full SHA for b42e77e

Commits on Dec 1, 2023

• wip checkpoint on tuning

  reubenharry

  committed
  2b323ce

  Copy full SHA for 2b323ce

• align blackjax and mclmc repos, for tuning

  reubenharry

  committed
  9a41cdf

  Copy full SHA for 9a41cdf

• use effective_sample_size

  reubenharry

  committed
  cdbb4f6

  Copy full SHA for cdbb4f6

• patial rename

  reubenharry

  committed
  947d717

  Copy full SHA for 947d717

• rename

  reubenharry

  committed
  e9ab7b4

  Copy full SHA for e9ab7b4

• clean up tuning

  reubenharry

  committed
  72d70c6

  Copy full SHA for 72d70c6

• clean up tuning

  reubenharry

  committed
  c121beb

  Copy full SHA for c121beb

Commits on Dec 2, 2023

• IN THIS COMMIT, BLACKJAX AND ORIGINAL REPO AGREE. SEED IS FIXED.

  reubenharry

  committed
  fe99163

  Copy full SHA for fe99163

• RANDOMIZE KEYS

  reubenharry

  committed
  c456efe

  Copy full SHA for c456efe

• ADD TEST

  reubenharry

  committed
  d0a008a

  Copy full SHA for d0a008a

• ADD TEST

  reubenharry

  committed
  d692498

  Copy full SHA for d692498

• Merge branch 'main' of https://github.com/blackjax-devs/blackjax

  reubenharry

  committed
  3e8d8ea

  Copy full SHA for 3e8d8ea

• Merge branch 'main' into refactor

  reubenharry

  committed
  eda029a

  Copy full SHA for eda029a

• MERGE MAIN

  reubenharry

  committed
  a45f58f

  Copy full SHA for a45f58f

• INCREASE CODE COVERAGE

  reubenharry

  committed
  2a21c56

  Copy full SHA for 2a21c56

• REMOVE REDUNDANT LINE

  reubenharry

  committed
  67f0de9

  Copy full SHA for 67f0de9

• ADD NAME 'mclmc'

  reubenharry

  committed
  3f55f5f

  Copy full SHA for 3f55f5f

• SPLIT KEYS AND FIX DOCSTRING

  reubenharry

  committed
  666c540

  Copy full SHA for 666c540

• FIX MINOR ERRORS

  reubenharry

  committed
  c1615f5

  Copy full SHA for c1615f5

• FIX MINOR ERRORS

  reubenharry

  committed
  ae1bf30

  Copy full SHA for ae1bf30

• Merge branch 'main' of https://github.com/blackjax-devs/blackjax

  reubenharry

  committed
  3c2dbad

  Copy full SHA for 3c2dbad

• FIX CONFLICT IN BIB

  reubenharry

  committed
  c396aa1

  Copy full SHA for c396aa1

• RANDOMIZE KEYS (reversion)

  reubenharry

  committed
  0902a1c

  Copy full SHA for 0902a1c

• PRECOMMIT CLEAN UP

  reubenharry

  committed
  2e3c80b

  Copy full SHA for 2e3c80b

Commits on Dec 3, 2023

• ADD KWARGS FOR DEFAULT HYPERPARAMS

  reubenharry

  committed
  604b5a9

  Copy full SHA for 604b5a9

Commits on Dec 4, 2023

• Merge branch 'main' of https://github.com/blackjax-devs/blackjax

  reubenharry

  committed
  50b1c95

  Copy full SHA for 50b1c95

• Merge branch 'main' into refactor

  reubenharry

  committed
  fecd82b

  Copy full SHA for fecd82b

Commits on Dec 5, 2023

• UPDATE ESS

  reubenharry

  committed
  50a8243

  Copy full SHA for 50a8243

• NAME CHANGES

  reubenharry

  committed
  a20a681

  Copy full SHA for a20a681

• NAME CHANGES

  reubenharry

  committed
  75e71de

  Copy full SHA for 75e71de

• MINOR FIXES

  reubenharry

  committed
  70f1dd5

  Copy full SHA for 70f1dd5