Home » Oreps » Jax » All Releases » jax-v0.4.9 Release

google/jax

Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more

Watch

Basic Details Issues Addressed Top Contributors Directory Browser

Jax jax-v0.4.9

Jax: jax-v0.4.9 Release

Release date:

May 9, 2023

Previous version:

jax-v0.4.9-rc (released May 9, 2023)

Magnitude:

1,058 Diff Delta

Contributors:

10 total committers

Data confidence:

Commits:

25 Commits in this Release

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

[shape_poly] Improve testing of vmap test harnesses

Authored May 9, 2023

Improve the shape incompatible error message by adding the argument/result name path to it.

Authored May 5, 2023

Fix the cache on `to_gspmd_sharding` to depend on if device/backend is set on pjit/jit.

Authored May 9, 2023

jax2tf: better handling for opaque dtypes

Authored May 4, 2023

Add McJAX resharding to device_put. Allow resharding if inputs and target sharding have the same set of devices but different order.

Authored May 9, 2023

delegate `select` lowering to opaque dtype rule

Authored May 9, 2023

Make `jax.jit` work with vmap(..., spmd_axis_name) when there is no mesh context manager.

Authored May 5, 2023

Improve the bisection strategy in eigh.

Authored May 5, 2023

Import "ir" directly rather than as "mlir.ir"

Authored May 9, 2023

KeyArray: support make_array_from_* APIs

Authored May 4, 2023

Fix the problem for tf function return StatefulPartitionedCall during jax2tf.call_tf.

Authored May 6, 2023

[GPU] Add missing stream synchronization to tridiagonal_solve gtsv2 call.

Authored May 8, 2023

Disable more tests that are flaky in CI.

Authored May 5, 2023

jnp.delete: avoid large trace-time constant when deleting slice

Authored May 5, 2023

Add docs on how to create a jax.Array from data parallel host local inputs

Authored May 5, 2023

improve serial_loop docstring

Authored May 5, 2023

Make `dimensions` argument of bcoo_reshape optional.

Authored May 9, 2023

Use stream-synchronized copy in rnn_kernels.cc.

Authored May 8, 2023

[GPU] Fix another instance of missing stream synchronization in RNN kernels.

Authored May 9, 2023

[shape_poly] Remove old code disabling some tests

Authored May 8, 2023

Revert setup.py changes.

Authored May 8, 2023

Update WORKSPACE and setup.py in preparation for jax/jaxlib 0.4.9 release, take 2

Authored May 9, 2023

Use set equality operators instead of intersection because I didn't know set had equality operators.

Authored May 9, 2023

[shape_poly] Add support for shape polymorphism with native serialization for lax.linalg.qr on TPU

Authored May 9, 2023

Add tests for negative inputs to top-k.

Authored May 8, 2023

Browse Other Releases

Latest Pending Unreleased 😎

jax-v0.4.11 Released May 31, 2023

0 Δ

jax-v0.4.11-rc Released May 31, 2023

6,799 Δ

jax-v0.4.10 Released May 11, 2023

0 Δ

jax-v0.4.10-rc Released May 11, 2023

1,233 Δ

jax-v0.4.9 Released May 9, 2023

1,058 Δ

jax-v0.4.9-rc Released May 9, 2023

23,581 Δ

jax-v0.4.8 Released March 29, 2023

2,231 Δ

jax-v0.4.7 Released March 27, 2023

0 Δ

jax-v0.4.7-rc1 Released March 27, 2023

0 Δ

jax-v0.4.7-rc Released March 27, 2023

19,399 Δ

Top Contributors in jax-v0.4.9

yashk2810

gnecula

jakevdp

hawkinsp

froystig

a-googler

maxwillzq

cgarciae

geraschenko

skye

Directory Browser for jax-v0.4.9

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