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

Loading