flax
Add flag for variables in nnx.merge
#5381
Merged

Add flag for variables in nnx.merge #5381

samanklesaria
samanklesaria samanklesaria requested a review from cgarciae cgarciae 114 days ago
samanklesaria samanklesaria force pushed from 64ffc0c0 to a554ad41 113 days ago
cgarciae
cgarciae approved these changes on 2026-03-31
cgarciae
cgarciae approved these changes on 2026-03-31
samanklesaria samanklesaria force pushed from 6875a598 to 5f4423fb 112 days ago
samanklesaria Add flag for variables in nnx.merge
148f202b
samanklesaria Rename to 'recreate_variables'
5716ff76
samanklesaria samanklesaria force pushed from 5f4423fb to 5716ff76 106 days ago
cgarciae
cgarciae approved these changes on 2026-04-07
cgarciae cgarciae added pull ready
copybara-service copybara-service merged 4c9d11fd into main 100 days ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
Milestone