jax-v0.10.2
📦 jaxView on GitHub →
✨ 4 features🔧 4 symbols
Summary
This release introduces several new functions in jax.scipy.linalg and adds a convenient shortcut to jax.ShapeDtypeStruct. Significant performance improvements were also made to JAX tracing.
✨ New Features
- Added jax.scipy.linalg.invhilbert for the closed-form inverse of the Hilbert matrix.
- Added jax.scipy.linalg.invpascal for the inverse of the Pascal matrix.
- Added jax.scipy.linalg.fiedler_companion for constructing the pentadiagonal Fiedler companion matrix of a polynomial.
- Added jax.ShapeDtypeStruct.like -- a shortcut for constructing a jax.ShapeDtypeStruct from an object with shape and dtype attributes.