Skip to content

Add jax array monoids and reduction rule#658

Open
jfeser wants to merge 3 commits intostaging-weightedfrom
jf-weighted-jax
Open

Add jax array monoids and reduction rule#658
jfeser wants to merge 3 commits intostaging-weightedfrom
jf-weighted-jax

Conversation

@jfeser
Copy link
Copy Markdown
Contributor

@jfeser jfeser commented May 6, 2026

Blockd by #655

* add monoid module

* clean up

* fix doctest

* fix

* wip

* remove incorrect rule

* add disjoint set tests and fix bug

* lint

* drop jax monoid defs

* drop incorrect comment

* add assert

* reduce nondeterminism and add assertions

* fix inconsistent stream numbering and missing constant factors
@jfeser jfeser changed the base branch from master to staging-weighted May 6, 2026 22:47
@jfeser jfeser force-pushed the staging-weighted branch from 7c22756 to d558162 Compare May 7, 2026 19:20
@jfeser jfeser force-pushed the jf-weighted-jax branch from fd924f2 to f7d43e5 Compare May 7, 2026 19:42
@jfeser jfeser added the blocked label May 8, 2026
@jfeser jfeser requested a review from eb8680 May 8, 2026 14:46
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant