flax
Return updates PyTree from nnx.Optimizer.update
#5214
Merged

Loading