Home » Oreps » Jax » All Releases » jax-v0.2.25 Release
google/jax
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
Watch
1
Commit Activity
Directory Browser
Velocity History
Releasesjax-v0.2.25
Basic Details Issues Addressed Top Contributors Directory Browser Release Notes
Jax jax-v0.2.25

Jax: jax-v0.2.25 Release

Release date:
October 25, 2021
Previous version:
jax-v0.2.24 (released October 19, 2021)
Magnitude:
2,029 Diff Delta
Contributors:
11 total committers
Data confidence:
Commits:
29 impacting commits

29 Commits in this Release

Ordered by the degree to which they evolved the repo in this version.

Move xla_bridge.constant to jax.interpreter.xla.pyval_to_ir_constant.
Authored October 19, 2021
[JAX] move example libraries from `jax.experimental` into `jax.example_libraries`
Authored October 20, 2021
[XLA:TPU] Adding approximate nearest neighbor search on TPU feature to JAX.
Authored October 19, 2021
special.lpmn: use more canonical testing approach
Authored October 19, 2021
add fft normalisation
Authored October 20, 2021
Implements padding support for lax.reduce_window when enable_xla=False.
Authored October 20, 2021
[XLA:TPU] Support jvp/vjp in approx_top_k
Authored October 20, 2021
stats.multivariate_normal: support broadcasted inputs
Authored October 19, 2021
Improve pytype checking of XLA types inside JAX.
Authored October 19, 2021
Fixes a dtype bug in the conversion of dynamic_slice when enable_xla=False.
Authored October 20, 2021
[JAX] Update users of `jax.ops.index...` functions, which are deprecated.
Authored October 19, 2021
[sparse] add .block_until_ready() to sparse objects
Authored October 21, 2021
jnp.histogramdd: more succinct density computation
Authored October 20, 2021
[sparse] improve error for BCOO.fromdense if nse is not specified
Authored October 20, 2021
Update jax2tf documentation with leading underscore when setting tf.Module() variables, e.g. `m._variables = ...`. Also added a test for this.
Authored October 20, 2021
[XLA:TPU] Fix approx_top_k output slice.
Authored October 19, 2021
Make the pjit docs clear about who does local and global communication
Authored October 25, 2021
odeint: validate *args
Authored October 20, 2021
Internal change
Authored October 21, 2021
Implements a lowering function for tf.expm1 and adds tests.
Authored October 22, 2021
Relax the tolerances in one 'lax_scipy_test' case
Authored October 20, 2021
odeint: args validation allows pytrees
Authored October 21, 2021
[JAX] Fix sphinx formatting issue
Authored October 21, 2021
Update jax version number after release
Authored October 19, 2021
Fix broken link to experimental/README.md
Authored October 20, 2021
[JAX] Replace users of deprecated jax.lib.xla_bridge functions with more public names.
Authored October 21, 2021
Reland: Use variadic reduce on GPU for argmax/argmin
Authored October 21, 2021
[JAX] Export ann documentation.
Authored October 20, 2021
DOC: add conv_dimension_numbers and ConvGeneralDilatedDimensionNumbers to docs
Authored October 20, 2021

Browse Other Releases

Latest Pending Unreleased 😎
jax-v0.3.1 Released February 18, 2022
6,721 Ξ”
jax-v0.3.0 Released February 10, 2022
5,978 Ξ”
jax-v0.2.28 Released February 2, 2022
829 Ξ”
jax-v0.2.27 Released January 18, 2022
6,502 Ξ”
jax-v0.2.26 Released December 8, 2021
1,528 Ξ”
jax-v0.2.25 Released October 25, 2021
2,029 Ξ”
jax-v0.2.24 Released October 19, 2021
298 Ξ”
jax-v0.2.23 Released October 19, 2021
0 Ξ”
jax-v0.2.22 Released October 13, 2021
37 Ξ”
jax-v0.2.21 Released September 23, 2021
10,380 Ξ”

Top Contributors in jax-v0.2.25

hawkinsp
jakevdp
a-googler
froystig
marcvanzee
iolloj
akolesnikoff
yashk2810
agrue
TheNeuralBit

Directory Browser for jax-v0.2.25

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

Release Notes Published

  • 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.
GitClear uses cookies to ensure you get the best experience on our website. Learn more