Skip to content

Add ImproperUniform#85

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

Add ImproperUniform#85
gvcallen wants to merge 1 commit into
lockwo:mainfrom
gvcallen:improper_uniform

Conversation

@gvcallen

@gvcallen gvcallen commented May 7, 2026

Copy link
Copy Markdown
Contributor

This is the canonical "Identity distribution". I've found that I've needed it as a placeholder many times when dealing with complex PyTrees.

For example, in my library Parax (which uses distreqx), you can attach distributions as metadata to PyTree nodes and then using tree_joint to extract the joint distribution over the resultant PyTree. Without an "identity" distribution, users cannot have raw JAX array's and Parax variables alongside.

Improper uniform distribution over the entire real line: unnormalized
density of 1 everywhere, so log_prob is always 0. Infinite entropy;
sampling, moments, and cdf-family methods are undefined and raise
NotImplementedError since the distribution doesn't integrate to 1.
@gvcallen gvcallen force-pushed the improper_uniform branch from 9dcfa11 to 323da46 Compare July 1, 2026 20:29
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