flax
feat: add `nnx.set_metadata` to in-place change metadata of the state variables of `nnx.Module`s
#5007
Merged
Merged
feat: add `nnx.set_metadata` to in-place change metadata of the state variables of `nnx.Module`s #5007
copybara-service
merged
1 commit into
google:main
from
pfackeldey:differentiable
cgarciae
commented
on 2025-10-08
pfackeldey
changed the title
pfackeldey
changed the title
vfdev-5
commented
on 2025-10-13
vfdev-5
approved these changes
on 2025-10-14
cgarciae
approved these changes
on 2025-10-14
feat: add nnx.set_metadata
a916d6c0
Login to write a write a comment.
Login via GitHub