flax
56216fa6 - Merge pull request #3983 from chiamp:optimizer

Commit

2 years ago

Merge pull request #3983 from chiamp:optimizer PiperOrigin-RevId: 642731140

References

#3983 - added `wrt` option to `nnx.Optimizer`

Author

a-googler

a-googler

Parents

Loading