Skip to content

v0.9.1

Choose a tag to compare

@github-actions github-actions released this 16 May 14:03
· 56 commits to main since this release

Fixed

  • MonadicProgram.log_joint rank-promotes upstream continuous values for single-arg steps. When a draw step's args was a single name, _resolve_input returned the env tensor verbatim, so a 1-D continuous draw (the shape Guide.rsample emits for a 1-D Euclidean codomain) fed nn.Linear as a (B,) row vector and crashed the downstream _NeuralSource matmul with mat1 and mat2 shapes cannot be multiplied (1xB and 1xH). Single-arg continuous inputs are now promoted to (B, 1) via the same rule _stack_tensors already applied for the multi-arg branch, with integer-dtype inputs (which feed nn.Embedding lookups) left untouched. _NeuralSource.forward also unsqueezes a stray rank-1 input as a belt-and-braces guard against custom ContinuousMorphisms regressing the same way. Affects every shipped Conditional* family sitting downstream of a 1-D continuous step; surfaced by direct MonadicProgram(steps=[...]) users on the first batched svi.step.

Full Changelog: v0.9.0...v0.9.1