Home » Oreps » Jax » All Releases » pending_release Release

google/jax

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

Watch

Commit Activity

Directory Browser

Velocity History

ReleasesLatest Pending

Basic Details Issues Addressed Top Contributors Directory Browser

Jax Latest Pending

Jax: Latest Pending Release

Release date:

Pending release in a future version

Previous version:

0.3.25 (released November 15, 2022)

Magnitude:

66,641 Diff Delta

Contributors:

96 total committers

Data confidence:

---

Commits:

674

674 Commits in this Release

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

Avoid imports from the public jax.* namespace in more places internally.

Authored April 4, 2023

[JAX] Improve handling of metadata in compilation cache.

Authored April 17, 2023

[jax2tf] First draft of supporting multi-platform serialization

Authored December 27, 2022

[jax2tf] Refactor jax2tf to move more pure-JAX stuff to jax_export

Authored April 4, 2023

Retrace jaxpr again if the same function with the same avals is run under different meshes. Add pxla.thread_resources to the global trace_context which takes care of the caching of global variables.

Authored March 2, 2023

`custom_vjp` symbolic zeros support, take two

Authored March 24, 2023

Bump minimum jaxlib version to 0.4.11. xla_extension_version is 158 and mlir_api_version is 49. It will subsume https://github.com/google/jax/pull/16161#issuecomment-1564977332

Authored June 1, 2023

YESSSS

Authored March 8, 2023

`custom_vjp` symbolic zeros support

Authored March 14, 2023

Checkify big refactor.

Authored May 23, 2023

Rename `in_axis_resources` and `out_axis_resources` with `in_shardings` and `out_shardings`. This is just a simple name replacement. It does not change any of the current pjit semantics and doesn't break any code.

Authored February 14, 2023

Preserve shardings on the output of pjit that were provided on the arguments.

Authored April 5, 2023

Switch jaxlib to use nanobind instead of pybind11.

Authored May 23, 2023

Add JAX pytree key path API to tree_utils.

Authored February 22, 2023

[jax2tf] Add backward compatibility tests for qr custom calls

Authored March 18, 2023

Improve the error message which is raised from `_get_and_check_device_assignment`.

Authored February 9, 2023

Make JAX-Triton calls serializable.

Authored May 17, 2023

Internal change

Authored March 4, 2023

[JAX] Delete _DeviceArray and DeviceArray.

Authored March 23, 2023

lax_test: Create a separate module for lax-specific test utils in a new package.

Authored February 6, 2023

Use `_to_xla_hlo_sharding` everywhere in JAX. Remove `_to_xla_op_sharding` in favor of `_to_xla_hlo_sharding` since constructing a C++ class is faster than protos and will help with further changes coming to HloSharding.

Authored May 27, 2023

Split Sharding subclasses out of _src/sharding.py into _src/sharding_impls.py

Authored March 13, 2023

Shorten alias chains for names exported in jax. namespace.

Authored February 25, 2023

Redefine the pile representation from concatenated to stacked-and-padded.

Authored May 5, 2023

[shape_poly] Add compile-time shape checking. Step 1.

Authored May 27, 2023

- Transitioning to new sharding controller

Authored November 20, 2022

Add support for dynamic shapes to jax.fft.

Authored April 2, 2023

[jax2tf] Add a backward compatibility test for tf.call_tf_function

Authored May 31, 2023

Merge device_put and with_sharding_constraint

Authored March 5, 2023

Support simple batching in dynamic_slice/update_slice

Authored April 13, 2023

[JAX] Add support for multiple pytree registries.

Authored May 25, 2023

Preserve shardings on the output of pjit that were provided on the arguments.

Authored April 5, 2023

[jax2tf] Add the first version of a custom call backwards compatibility test

Authored March 10, 2023

[jax2tf] Add a backward compatibility test for tf.call_tf_function

Authored June 1, 2023

Add a debug module to dump stack traces in a file explicitly on a fault or after a user signal.

Authored April 10, 2023

Split Mesh and ResourceEnv into a new module jax._src.mesh.

Authored February 27, 2023

Fixed loc and scale parameters for logistic distribution. CDF and SF have been added for several distributions, including cauchy, gamma, logistic, chi2 and beta. ISF and PPF have also been added for cauchy and logistic.

Authored March 5, 2023

Update api_benchmark to not use any deprecated APIs.

Authored February 28, 2023

[jax2tf] Refactor the gradient machinery for native serialization

Authored April 4, 2023

improve vmap pytree error messages

Authored March 2, 2023

Browse Other Releases

Latest Pending Unreleased 😎

0.3.25 Released November 15, 2022

6,881 Δ

0.3.24 Released November 4, 2022

23,075 Δ

0.3.23 Released October 12, 2022

305 Δ

0.3.22 Released October 11, 2022

11,637 Δ

0.3.21 Released September 30, 2022

1,561 Δ

0.3.20 Released May 31, 2023

152,465 Δ

jaxlib-v0.4.11 Released May 31, 2023

6,797 Δ

jaxlib-v0.4.10 Released May 11, 2023

1,233 Δ

jaxlib-v0.4.9 Released May 9, 2023

25,433 Δ

jaxlib-v0.4.7 Released March 27, 2023

19,058 Δ

Top Contributors in Latest Pending

gnecula

hawkinsp

yashk2810

jakevdp

mattjj

pschuh

a-googler

chrisflesher

froystig

LenaMartens

Directory Browser for Latest Pending

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