Add jax dispatch for `KroneckerProduct` `Op`#822

Merged

jessegrabowski merged 1 commit intopymc-devs:mainfrom

jessegrabowski:jax-kron

Jun 15, 2024

Commits on Jun 15, 2024

Add jax dispatch for `KroneckerProduct` `Op`
jessegrabowski
committed