jaxlib version 0.3.15

jaxlib-v0.3.15

jaxlib version 0.3.15

jax version 0.3.15

jax-v0.3.15

jax version 0.3.15

Jaxlib release v0.3.14

jaxlib-v0.3.14

Jaxlib release v0.3.14

JAX release v0.3.14

jax-v0.3.14

JAX release v0.3.14

JAX release v0.3.13

jax-v0.3.13

jax version 0.3.13

Jax release v0.3.12

• Changes
  • Fixes #10717

Jax release v0.3.11

• 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.

Jaxlib release v0.3.10

Update TF commit for release

PiperOrigin-RevId: 446555288

Jax release v0.3.10

Update TF commit for release

PiperOrigin-RevId: 446555288

Jax release 0.3.9

• Changes
  • Added support for fully asynchronous checkpointing for GlobalDeviceArray.