flax
87d91567 - ModelParamTraversal for linen

Commit
5 years ago
ModelParamTraversal for linen allows ModelParamTraversal to operate on dict/FrozenDict, while keeping backward compat with Model instances
Author
Committer
Parents
Loading