flax
a916d6c0
- feat: add nnx.set_metadata
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
81 days ago
feat: add nnx.set_metadata
References
#5007 - feat: add `nnx.set_metadata` to in-place change metadata of the state variables of `nnx.Module`s
Author
pfackeldey
Parents
75fd8fa1
Loading