Staging branch for `weighted` - #656

Draft

jfeser wants to merge 18 commits into

staging-weighted

Commits on Aug 10, 2026

Add monoid module (#653)
jfeser
committed
bf822e8
Add inversion from `weighted` (#655)
jfeser
committed
3e06469
Refactor `monoid.py` to remove class structure (#661)
jfeser
committed
75cf1ef
Add `jax` array monoids and reduction rule (#658)
jfeser
committed
d46494b
Add `delta` terms for array construction in `handlers.jax.monoid` (#663)
jfeser
committed
8f851f7
Add weighted streams (#665)
jfeser
committed
cd7ae91
Replace factorization rules with a push-based rule (#672)
jfeser
committed
3c22d40
Add weighted einsum implementation (#671)
jfeser
committed
fa11a88
Break ReduceArrayGather rule into 2 steps (#681)

eb8680
authored and
jfeser
committed
8c4fea6
Remove dead code (#683)

eb8680
authored and
jfeser
committed
5da1b52
Implement plated einsum (#692)
jfeser
committed
398b41b
Monoid.reduce(b, {}) == b (#723)
jfeser
committed
48afbeb
Don't route all-scalar monoid ops through jax (#729)

authored and
jfeser
committed
e0488da
Add a generator-expression bytecode disassembler (#725)

authored and
jfeser
committed
c646484
fix test harness (#740)
jfeser
committed
c53a0e0
asarray takes both named and positional arguments (#741)
jfeser
committed
7e0dbb5
Support reduction over distributions with enumerable support (#737)
jfeser
committed
19e211f

Commits on Aug 12, 2026

Add `Group` and associated rules (#752)
eb8680
authored
58c694a