Return updates PyTree from nnx.Optimizer.update #5214
anshuljoshi
wants to merge
1 commit into
google:main
from
anshuljoshi:optimizer-return-updates
Login to write a write a comment.
Login via GitHub