Skip to content

Implement support for product kernels - #10

Merged
evan-palmer merged 2 commits into
mainfrom
dev-product-kernel
Nov 4, 2026
Merged

evan-palmer merged 2 commits into
mainfrom
dev-product-kernel

Conversation

@evan-palmer

Copy link
Copy Markdown
Member

Changes Made

This PR makes two changes:

  1. It adds a wrapper for generating product kernels from a pytree of kernels
  2. It phases out the old jax.tree_util namespace with the current jax.tree API

@evan-palmer evan-palmer self-assigned this Nov 4, 2026
@evan-palmer
evan-palmer merged commit 164d455 into main Nov 4, 2026
3 checks passed
@evan-palmer
evan-palmer deleted the dev-product-kernel branch November 4, 2026 20:13
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