Home » Oreps » Jax » All Releases » jax-v0.4.12-rc 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.4.12-rc
Basic Details Issues Addressed Top Contributors Directory Browser
Jax jax-v0.4.12-rc

Jax: jax-v0.4.12-rc Release

Release date:
June 8, 2023
Previous version:
jax-v0.4.11 (released May 31, 2023)
Magnitude:
4,680 Diff Delta
Contributors:
21 total committers
Data confidence:
Commits:
79 impacting commits

79 Commits in this Release

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

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
Redefine the pile representation from concatenated to stacked-and-padded.
Authored May 5, 2023
[jax2tf] Add a backward compatibility test for tf.call_tf_function
Authored June 1, 2023
Allow GlobalAsyncCheckpointManager to work on a single process without initializing the distributed system.
Authored June 5, 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 June 5, 2023
Convert OpSharding to HloSharding in the constructor of GSPMDSharding. Also make `op_sharding_to_indices` to work with only HloSharding.
Authored June 6, 2023
[shape_poly] Keep track of whether a lowering contains shape polymorphism
Authored May 31, 2023
Clean up old jaxlib version guards.
Authored June 6, 2023
Add support for dtype float8_e4m3b11fnuz
Authored June 6, 2023
Allow duck-typed inputs to zeros_like, ones_like, etc.
Authored June 6, 2023
[shape_poly] Add a polymorphic shape refinement MLIR pass accessible to JAX Python.
Authored June 3, 2023
jnp.roll: more efficient implementation for static shifts
Authored June 2, 2023
Make `inspect_sharding_lowering_rule` work only with HloSharding in it's callback. Also remove `get_replicated_op_sharding` since it is not needed anymore.
Authored June 7, 2023
custom prng: add shard methods to PRNGKeyArrayImpl
Authored June 1, 2023
[shape_poly] Enable calling from JAX modules that use shape polymorphism.
Authored June 6, 2023
fix checkify + custom_vjp after symbolic zeros change
Authored June 7, 2023
Improve printing of bints and piles, and allow bints in convert_element_type.
Authored May 18, 2023
jax.nn.softmax: fix fill value when where is specified
Authored June 1, 2023
Update profiling docs.
Authored May 30, 2023
Fix `get_tensorstore_spec` for GCS paths if ocdbt is enabled
Authored June 7, 2023
Simplify sharding types input to physical_hlo_sharding and lower_jaxpr_to_fun.
Authored June 7, 2023
Mark jax.abstract_arrays as deprecated
Authored June 6, 2023
Fix test_util for new float8 type
Authored June 8, 2023
Avoid IndexError when constructing a ValueError for a DeviceAssignmentMismatchError.
Authored June 2, 2023
Deprecate jax.numpy functions alltrue, sometrue, product, cumproduct
Authored June 2, 2023
fix key opaque dtype physical HLO sharding handler
Authored June 6, 2023
Simplify `are_op_shardings_equal` a bit so that some vertical space is reduced
Authored June 6, 2023
Add a test
Authored May 19, 2023
cleanup old jaxlib version check
Authored June 6, 2023
Remove some dead code (followup to #16159)
Authored May 28, 2023
Raise a better error.
Authored April 17, 2023
Tidy up in response to first review.
Authored June 3, 2023
Add .editorconfig
Authored June 2, 2023
revise vmap's axis_frame-getting logic
Authored May 24, 2023
Internal change
Authored June 1, 2023
Change the _most_recent_executable logic to store a weakref dict of jaxpr -> executable so that with the inner cpp cache and outer cpp cache, we extract the correct executable.
Authored June 5, 2023
Delete OpShardingSharding export since it has been 3 months since it was deprecated. Also remove deprecation warnings for MeshPspecSharding.
Authored June 8, 2023
Force the mask computation into 'int32' more consistently
Authored May 19, 2023
custom_parititioning: in lower sharding, Sharding should be XLACompatibleSharding.
Authored June 1, 2023
Add a comment at the end of autotuning a Triton function
Authored June 7, 2023

Browse Other Releases

Latest Pending Unreleased 😎
jax-v0.4.14-rc Released July 27, 2023
19,891 Ξ”
jax-v0.4.13 Released June 22, 2023
0 Ξ”
jax-v0.4.13-rc Released June 22, 2023
7,868 Ξ”
jax-v0.4.12 Released June 8, 2023
0 Ξ”
jax-v0.4.12-rc Released June 8, 2023
4,680 Ξ”
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 Ξ”

Top Contributors in jax-v0.4.12-rc

yashk2810
jakevdp
gnecula
axch
mattjj
nouiz
skye
andresusanopinto
hawkinsp
froystig

Directory Browser for jax-v0.4.12-rc

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

GitClear uses cookies to ensure you get the best experience on our website. Learn more