What's Changed
- Add rewrites to remove unnecessary
expm1
operations by @ricardoV94 in #759 - Fix
DiffOp
output type when input has partially known shape by @ricardoV94 in #855 - Make
Sparse
tensorType
s extendTensorType
by @aerubanov in #766 - Use new JAX index update approach when available by @brandonwillard in #864
- Convert shapes to scalar tuples in the Numba
BroadcastTo
implementation by @brandonwillard in #863
Full Changelog: rel-2.5.1...rel-2.5.2