Skip to content

Add Leafwise bijector#83

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

Add Leafwise bijector#83
gvcallen wants to merge 1 commit into
lockwo:mainfrom
gvcallen:treemap

Conversation

@gvcallen

@gvcallen gvcallen commented Apr 30, 2026

Copy link
Copy Markdown
Contributor

Adds a Leafwise bijector, analogous to tensorflow's JointMap and opposite to Chain.

This applies a tree of bijectors to a tree of inputs.

@gvcallen gvcallen changed the title Add TreeMap Add Leafwise May 8, 2026
@gvcallen gvcallen changed the title Add Leafwise Add Leafwise bijector May 8, 2026
Leafwise applies a tree of bijectors to a matching tree of inputs,
analogous to TensorFlow's JointMap and the pytree-level opposite of
Chain (which composes bijectors sequentially on a single input).
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