rel-2.5.2

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 tensor Types extend TensorType 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