flax
Fix bound method auto-unbinding for NNX transforms
#5055

Merged

Fix bound method auto-unbinding for NNX transforms #5055

copybara-service merged 5 commits into google:main from mohsinm-dev:fix-bound-method-auto-unbinding

Fix bound method auto-unbinding for NNX transforms

9d453659

samanklesaria commented on 2025-10-28

Replace bound method auto-unbinding with error detection

fa36a607

samanklesaria commented on 2025-10-30

Remove unused bound method handling code

e5e87e4a

samanklesaria requested a review from

vfdev-5 145 days ago

Run pre-commit hook

10c2da40

cgarciae approved these changes on 2025-11-06

cgarciae added pull ready

Merge branch 'main' into pr/5055

dece6b34

copybara-service merged d6e8e14f into main 127 days ago

Reviewers

cgarciae

samanklesaria

vfdev-5

Assignees

No one assigned

Labels

pull ready

Milestone

No milestone