flax
54fe15f2 - Merge pull request #4007 from google:nnx-fix-grad

Commit

2 years ago

Merge pull request #4007 from google:nnx-fix-grad PiperOrigin-RevId: 645050031

References

#4007 - [nnx] fix grad

Author

a-googler

a-googler

Parents

Loading