flax
9a40b3fa - added wrt option to nnx.Optimizer.update

Commit

1 year ago

added wrt option to nnx.Optimizer.update

References

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

Author

chiamp

chiamp

Committer

chiamp

chiamp

Parents

Loading