jax
0f103d33 - Merge pull request #22836 from superbobry:maint-2

Commit

1 year ago

Merge pull request #22836 from superbobry:maint-2 PiperOrigin-RevId: 659644462

References

#22836 - Added a custom JVP rule for jax.nn.logsumexp

Author

a-googler

a-googler

Parents

Loading