Skip to content

JAX release v0.2.25
Compare
Choose a tag to compare
@zhangqiaorjc zhangqiaorjc released this 10 Nov 22:26
· 11219 commits to main since this release
jax-v0.2.25
821fcaa

  • New features:

    • (Experimental) jax.distributed.initialize exposes multi-host GPU backend.
    • jax.random.permutation supports new independent keyword argument
      ({jax-issue}#8430)

  • Breaking changes

    • Moved jax.experimental.stax to jax.example_libraries.stax
    • Moved jax.experimental.optimizers to jax.example_libraries.optimizers

  • New features:

    • Added jax.lax.linalg.qdwh.
Assets 2