flax
added `wrt` option to `nnx.Optimizer`
#3983
Merged

Commits
  • added wrt option to nnx.Optimizer.update
    chiamp committed 1 year ago
Loading