flax
e2c3dfd3 - Merge pull request #3463 from levskaya:vjp_fix

Commit

2 years ago

Merge pull request #3463 from levskaya:vjp_fix PiperOrigin-RevId: 579977891

References

#3463 - Add simple argument-only lifted nn.grad function.

Author

a-googler

a-googler

Parents

Loading