filter grad state in nnx.Optimizer - SemanticDiff

Commit

64 days ago

filter grad state in nnx.Optimizer PiperOrigin-RevId: 823228572

References

#5049 - filter grad state in nnx.Optimizer

Author

Cristian Garcia

Committer

a-googler

a-googler

Parents

Loading