fix(jax): route concrete kwarg attrs to aux_data in pytree flatten#1221

Merged

Jammy2211 merged 1 commit intomainfrom

feature/pixelization-pytree-migration

Apr 17, 2026

Commits on Apr 17, 2026

fix(jax): route concrete kwarg attrs to aux_data in pytree flatten

authored and
Jammy2211
committed