Code

Jittable transforms#1575

Code

Merged

fehiepsi merged 15 commits intopyro-ppl:masterpyro-ppl/numpyro:masterfrom pierreglaser:jittable-transformspierreglaser/numpyro:jittable-transformsCopy head branch name to clipboardMay 31, 2023

+656-23

Commits

Commits on Apr 11, 2023

• [WIP] jittable transforms

  pierreglaser

  committed
  f712984

  Copy full SHA for f712984

• add licence to new test file

  pierreglaser

  committed
  0ca2586

  Copy full SHA for 0ca2586

Commits on Apr 16, 2023

• turn BijectorConstraint into pytree

  pierreglaser

  committed
  d857ca4

  Copy full SHA for d857ca4

• test flattening/unflattening of parametrized constraints

  pierreglaser

  committed
  34e5dac

  Copy full SHA for 34e5dac

• cosmetic edits

  pierreglaser

  committed
  bc2d097

  Copy full SHA for bc2d097

• fix typo

  pierreglaser

  committed
  e9e6cb9

  Copy full SHA for e9e6cb9

• implement tree_flatten/unflatten for transforms

  pierreglaser

  committed
  ef486a0

  Copy full SHA for ef486a0

• attempt to avoid confusing black

  pierreglaser

  committed
  d2007f9

  Copy full SHA for d2007f9

• add (un)flattening meths for BijectorTransform

  pierreglaser

  committed
  e0a2b30

  Copy full SHA for e0a2b30

Commits on Apr 17, 2023

• fixup! implement tree_flatten/unflatten for transforms

  pierreglaser

  committed
  48f3b2e

  Copy full SHA for 48f3b2e

Commits on May 25, 2023

• test vmapping over transforms/constraints

  pierreglaser

  committed
  c468f04

  Copy full SHA for c468f04

• Merge branch 'master' into jittable-transforms

  pierreglaser

  committed
  8ba7767

  Copy full SHA for 8ba7767

Commits on May 30, 2023

• Make constraints __eq__ checks robust to arbitrary inputs

  pierreglaser

  committed
  49b051c

  Copy full SHA for 49b051c

• make transforms equality check robust to arbitrary inputs

  pierreglaser

  committed
  d1c4d71

  Copy full SHA for d1c4d71

• test constraints and transforms equality checks

  pierreglaser

  committed
  9584a4a

  Copy full SHA for 9584a4a