Skip to content

fix: traced instance attributes never land in pytree aux data - #1441

Merged
Jammy2211 merged 1 commit into
mainfrom
feature/jax-pytree-traced-aux-fix
Aug 1, 2026
Merged

fix: traced instance attributes never land in pytree aux data#1441
Jammy2211 merged 1 commit into
mainfrom
feature/jax-pytree-traced-aux-fix

Commits

Commits on Aug 1, 2026