Skip to content

fix: don't register JAX-leaf types (int) as pytree nodes (shapelet vmap crash)#1366

Merged
Jammy2211 merged 1 commit into
mainfrom
feature/jax-pytree-leaf-registration
Jul 13, 2026
Merged

fix: don't register JAX-leaf types (int) as pytree nodes (shapelet vmap crash)#1366
Jammy2211 merged 1 commit into
mainfrom
feature/jax-pytree-leaf-registration

Commits

Commits on Jul 13, 2026