flax
[flax] unconditionally register nnx.Variable as a pytree
#4748
Merged

[flax] unconditionally register nnx.Variable as a pytree #4748

copybara-service merged 1 commit into main from test_758800900
copybara-service
copybara-service copybara-service force pushed from ad1921cb to 18549434 1 year ago
[flax] unconditionally register nnx.Variable as a pytree
866fe952
copybara-service copybara-service force pushed from 18549434 to 866fe952 1 year ago
copybara-service copybara-service merged 866fe952 into main 1 year ago
copybara-service copybara-service deleted the test_758800900 branch 1 year ago

Login to write a write a comment.

Login via GitHub

Reviewers
No reviews
Assignees
No one assigned
Labels
Milestone