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

fix: don't register JAX-leaf types (int) as instance pytree nodes

2da8157
Select commit
Loading
Failed to load commit list.

Select a check to view from the sidebar