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

Commit

1 year 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