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

Jax: jaxlib-v0.4.12 Release

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

Browse Other Releases

Latest Pending Unreleased 😎
jaxlib-v0.4.16 Released September 18, 2023
10,382 Ξ”
jaxlib-v0.4.15 Released August 30, 2023
15,872 Ξ”
jaxlib-v0.4.14 Released July 27, 2023
17,830 Ξ”
jaxlib-v0.4.13 Released June 22, 2023
7,284 Ξ”
jaxlib-v0.4.12 Released June 8, 2023
4,131 Ξ”
jaxlib-v0.4.11 Released June 3, 2023
6,761 Ξ”
jaxlib-v0.4.10 Released June 3, 2023
1,137 Ξ”
jaxlib-v0.4.9 Released June 3, 2023
24,025 Ξ”
jaxlib-v0.4.7 Released March 27, 2023
25,002 Ξ”
jaxlib-v0.4.6 Released March 9, 2023
19,153 Ξ”

Top Contributors in jaxlib-v0.4.12

yashk2810
jakevdp
gnecula
axch
mattjj
nouiz
skye
hawkinsp
andresusanopinto
froystig

Directory Browser for jaxlib-v0.4.12

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 about our cookie policy