flax
Allow `nnx.bridge.variables.nnx_attrs_to_linen_vars` take `nnx.VariableState` as argument.
#4473

Merged

Allow `nnx.bridge.variables.nnx_attrs_to_linen_vars` take `nnx.VariableState` as argument. #4473

copybara-service merged 1 commit into main from test_713442378

copybara-service force pushed 1 year ago

Allow `nnx.bridge.variables.nnx_attrs_to_linen_vars` take `nnx.Variab…

9ebdbdc2

copybara-service force pushed to 9ebdbdc2 1 year ago

copybara-service merged 9ebdbdc2 into main 1 year ago

copybara-service deleted the test_713442378 branch 1 year ago

Reviewers

No reviews

Assignees

No one assigned

Labels

None yet

Milestone

No milestone