Skip to content

Fix Transformed/Chain for pytree-valued bijectors#92

Open
gvcallen wants to merge 2 commits into
lockwo:mainfrom
gvcallen:transformed_pytree_shapes
Open

Fix Transformed/Chain for pytree-valued bijectors#92
gvcallen wants to merge 2 commits into
lockwo:mainfrom
gvcallen:transformed_pytree_shapes

Updated to use jax.tree.map

4a1d22f
Select commit
Loading
Failed to load commit list.
Sign in for the full log view

Annotations

1 warning
run-test (3.10, ubuntu-latest)
succeeded Jul 7, 2026 in 1m 43s