flax
917c0974 - Add support for storing arbitrary PyTrees with `Module.perturb()`

Commit

1 year ago

Add support for storing arbitrary PyTrees with `Module.perturb()` PiperOrigin-RevId: 691525475

Author

duckworthd

Committer

a-googler

Parents