Code

Numpyro#122

Code

Merged

daniel-dodd merged 16 commits intov0.5_updateJaxGaussianProcesses/GPJax:v0.5_updatefrom numpyroJaxGaussianProcesses/GPJax:numpyroCopy head branch name to clipboardOct 16, 2022

+417-385

Commits

Commits on Oct 12, 2022

• Distrax to numpyro

  thomaspinder

  committed
  51fd1e2

• Resolve conflicts

  thomaspinder

  committed
  4e7420c

• Merge branch 'v0.5_update' of github.com:thomaspinder/GPJax into numpyro

  thomaspinder

  committed
  ed8f74e

Commits on Oct 13, 2022

• Update numpyro

  thomaspinder

  committed
  11b0d9e

Commits on Oct 14, 2022

• Improve kl divergence efficiency.

  daniel-dodd

  committed
  b40f883

• Fix variational expectation vmap.

  daniel-dodd

  committed
  70a400e

Commits on Oct 15, 2022

• Fix dense kl bug.

  daniel-dodd

  committed
  9c1607c

• Fix bijection bug.

  daniel-dodd

  committed
  c1eb346

• Numpyro mwe

  thomaspinder

  committed
  709c440

• Update workflow

  thomaspinder

  committed
  171fdd9

• Drop distrax ref

  thomaspinder

  committed
  e96b918

• Drop TFP dependency

  thomaspinder

  committed
  1252d46

• Update docs

  thomaspinder

  committed
  df530f6

• Remove Distrax from notebooks

  thomaspinder

  committed
  0d73ef8

• Fix missing dep

  thomaspinder

  committed
  e84489a

Commits on Oct 16, 2022

• Resolve review comments

  thomaspinder

  committed
  40ffc83