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

fix: traced instance attributes never land in pytree aux data

9283d2d
Select commit
Loading
Failed to load commit list.