flax
e4dad9ca - Merge pull request #4306 from google:nnx-custom-vjp-issue

Commit

1 year ago

Merge pull request #4306 from google:nnx-custom-vjp-issue PiperOrigin-RevId: 691392585

References

Author

a-googler

Parents