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

Jax: jaxlib-v0.4.22 Release

Release date:
December 13, 2023
Previous version:
jaxlib-v0.4.21 (released December 4, 2023)
Magnitude:
7,566 Diff Delta
Contributors:
20 total committers
Data confidence:
Commits:
102 impacting commits

102 Commits in this Release

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

Upgrade JAX pytrees doc
Authored December 7, 2023
Upgrade JAX Installation doc
Authored December 4, 2023
Upgrade most .py sources to 3.9
Authored December 8, 2023
[export] Add support for serialization and deserialization of Exported
Authored December 12, 2023
Add jax.extend.mlir.
Authored December 6, 2023
Upgrade JAX debugging doc
Authored December 11, 2023
[shape_poly] Add simpler APIs max_dim and min_dim, improve >= 0
Authored December 7, 2023
[Pallas/TPU] Add all gather kernel example
Authored December 5, 2023
[XLA:Mosaic] Support expanding lane dim in shapecast: (..., 128) -> (..., m * 128) and handle relayout from (1, 128) to (8, 128) for more general cases.
Authored December 5, 2023
Use an explicit MLIR dialect registration, rather than _site_initialize_0.
Authored December 6, 2023
[bazel] Add a BUILD file for jax/extend, and add more granular targets for individual pieces of extend.
Authored December 5, 2023
jnp.nonzero: deprecate zero-dimensional inputs
Authored December 6, 2023
[Pallas/Mosaic] Fixes for interpret mode on TPU
Authored December 6, 2023
Add a registry for primitives that require device_assignment during lowering
Authored December 9, 2023
Fix cache_used metric implementation and test.
Authored December 8, 2023
increase sample sizes on noisy random tests
Authored December 12, 2023
Fix the pjit flakey test. The test was weirdly written in the first place. The current suspicion is that `x.copy()` inside jit made it flakey. Also delete some duplicate tests from the time of migrating to jax.Array
Authored December 13, 2023
[bazel] Add a macro if_building_jaxlib() to guard dependencies that should only be present if building jaxlib.
Authored December 5, 2023
Improve shape validation when jax_dynamic_shapes=True
Authored December 12, 2023
[XLA:Mosaic] Support tpu.bitcast for i16, i8.
Authored December 11, 2023
`mesh_utils.create_hybrid_device_mesh`: make sorting granules by key user configurable.
Authored December 12, 2023
remove upstream-nightly CI job
Authored December 8, 2023
Replace device_buffers with addressable_shards in test because device_buffers is deprecated
Authored December 4, 2023
introduce a config flag to control a random seed offset
Authored December 13, 2023
[Pallas/Mosaic] Lower `lax.fori_loop`s to *rolled* loops.
Authored December 8, 2023
Add support for unrolling to `lax.fori_loop`
Authored December 4, 2023
improve a debug.callback type error message for idiots
Authored December 6, 2023
Disable all_gather_test on non-v5e TPUs
Authored December 7, 2023
Disable sanitizer builds of shape_poly_test.
Authored December 4, 2023
improve an error message during Mesh creation
Authored December 7, 2023
Fix indexing with slices when the slice elements are jax.Array.
Authored December 5, 2023
jaxpr: improve printed repr when eqn has no return values
Authored December 6, 2023
[shard-map] fix varargs error message bug
Authored December 13, 2023
array api: add jnp.linalg.cross & jnp.linalg.outer
Authored December 12, 2023
must flatten defaultdict in key-sorted order, like regular dicts
Authored December 8, 2023
Move jit to the callsite.
Authored December 9, 2023
[shard-map] register cumsum et al with generic rules
Authored December 13, 2023
remove _use_xeinsum from jnp.einsum api
Authored December 9, 2023
Print pjit name= before other params
Authored December 7, 2023
Reverts 3c07c10a9a55f9a32390dd10cf3f420bdf3f1ed8
Authored December 13, 2023

Browse Other Releases

Latest Pending Unreleased 😎
0.3.20 Released February 5, 2024
132,070 Ξ”
jaxlib-v0.4.26 Released April 3, 2024
16,870 Ξ”
jaxlib-v0.4.25 Released February 24, 2024
17,299 Ξ”
jaxlib-v0.4.24 Released February 5, 2024
29,867 Ξ”
jaxlib-v0.4.23 Released December 14, 2023
930 Ξ”
jaxlib-v0.4.22 Released December 13, 2023
7,566 Ξ”
jaxlib-v0.4.21 Released December 4, 2023
15,541 Ξ”
jaxlib-v0.4.20 Released November 2, 2023
5,123 Ξ”
jaxlib-v0.4.19 Released October 18, 2023
5,722 Ξ”
jaxlib-v0.4.18 Released October 6, 2023
3,149 Ξ”

Top Contributors in jaxlib-v0.4.22

8bitmp3
hawkinsp
gnecula
sharadmv
jakevdp
superbobry
yashk2810
mattjj
bythew3i
a-googler

Directory Browser for jaxlib-v0.4.22

All files are compared to previous version, jaxlib-v0.4.21. Click here to browse diffs between other versions.

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