Code

More bugfixes for statespace#346

Code

Merged

jessegrabowski merged 13 commits intopymc-devs:mainpymc-devs/pymc-extras:mainfrom

jessegrabowski:notebook-updatejessegrabowski/pymc-extras:notebook-updateCopy head branch name to clipboard

Jun 29, 2024

Commits

Commits on Jun 28, 2024

• Allow forward sampling of statespace models in JAX mode

  Show description for 4007c18
  jessegrabowski

  committed
  4007c18

  Copy full SHA for 4007c18

• Adjust test suite to reflect API changes

  Show description for 40b5871
  jessegrabowski

  committed
  40b5871

  Copy full SHA for 40b5871

• Add JAX test suite

  jessegrabowski

  committed
  49bd43a

  Copy full SHA for 49bd43a

• Bug-fixes and changes to statespace distributions

  Show description for feaadd0
  jessegrabowski

  committed
  feaadd0

  Copy full SHA for feaadd0

• Re-run example notebooks

  jessegrabowski

  committed
  3a72872

  Copy full SHA for 3a72872

• Add helper function to sample prior/posterior statespace matrices

  jessegrabowski

  committed
  09b6b18

  Copy full SHA for 09b6b18

• fix tests

  jessegrabowski

  committed
  e95a670

  Copy full SHA for e95a670

• Wrap jax MvNormal rewrite in try/except block

  jessegrabowski

  committed
  b7304f6

  Copy full SHA for b7304f6

• Don't use `action` keyword in `catch_warnings`

  jessegrabowski

  committed
  064af80

  Copy full SHA for 064af80

• Skip JAX test if `numpyro` is not installed

  jessegrabowski

  committed
  ae3ecd1

  Copy full SHA for ae3ecd1

Commits on Jun 29, 2024

• Handle batch dims on `SequenceMvNormal`

  jessegrabowski

  committed
  69b5b17

  Copy full SHA for 69b5b17

• Remove unused batch_dim logic in SequenceMvNormal

  jessegrabowski

  committed
  48bf59f

  Copy full SHA for 48bf59f

• Restore `get_support_shape_1d` import

  jessegrabowski

  committed
  74f52ac

  Copy full SHA for 74f52ac