Skip to content
This repository has been archived by the owner on May 11, 2023. It is now read-only.

Metadata via trace #1

Merged
merged 2 commits into from
Mar 22, 2023
Merged

Metadata via trace #1

merged 2 commits into from
Mar 22, 2023

Conversation

daniel-dodd
Copy link
Member

  • New recursive trace structure that does not use jax.tree_util.flatten!
  • Tests for the bijectors.

@daniel-dodd daniel-dodd merged commit b397e97 into main Mar 22, 2023
@daniel-dodd daniel-dodd deleted the metadata_via_trace branch March 22, 2023 14:55
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

1 participant