Skip to content

Add Deterministic and Joint#70

Open
gvcallen wants to merge 1 commit into
lockwo:mainfrom
gvcallen:deterministic_and_joint
Open

Add Deterministic and Joint#70
gvcallen wants to merge 1 commit into
lockwo:mainfrom
gvcallen:deterministic_and_joint

Conversation

@gvcallen

@gvcallen gvcallen commented Apr 4, 2026

Copy link
Copy Markdown
Contributor

Adds a Deterministic (point mass) and Joint distribution over pytrees. Reference code is from distrax (Deterministic and Joint)

@gvcallen gvcallen changed the title Add deterministic and joint distributions Add Deterministic and Joint Apr 4, 2026
Deterministic places all probability mass on a single point (with an
atol/rtol tolerance window for closeness comparisons). Joint composes a
pytree of statistically independent distributions, summing
log-probabilities/entropies/KL divergences over the tree. Ported from
distrax as reference.
@gvcallen gvcallen force-pushed the deterministic_and_joint branch from c177f9a to c88707d Compare July 1, 2026 20:43
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant