Jax: jax-v0.3.11 Release

Release date:
May 14, 2022
Previous version:
jax-v0.3.10 (released May 4, 2022)
Magnitude:
25 Diff Delta
Contributors:
1 total committer
Data confidence:
Commits:

Top Contributors in jax-v0.3.11

YouJiacheng

Directory Browser for jax-v0.3.11

We haven't yet finished calculating and confirming the files and directories changed in this release. Please check back soon.

Release Notes Published

  • Changes
    • {func}jax.lax.eigh now accepts an optional sort_eigenvalues argument that allows users to opt out of eigenvalue sorting on TPU.
  • Deprecations
    • Non-array arguments to functions in {mod}jax.lax.linalg are now marked keyword-only. As a backward-compatibility step passing keyword-only arguments positionally yields a warning, but in a future JAX release passing keyword-only arguments positionally will fail. However, most users should prefer to use {mod}jax.numpy.linalg instead.
    • {func}jax.scipy.linalg.polar_unitary, which was a JAX extension to the scipy API, is deprecated. Use {func}jax.scipy.linalg.polar instead.