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

Jax: jax-v0.4.22-rc Release

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

100 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
[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
Import new version of Triton
Authored December 13, 2023
Skip DLPack tests that are failing with the recent changes to the C API
Authored December 11, 2023

Browse Other Releases

Latest Pending Unreleased 😎
jax-v0.4.24-rc Released February 5, 2024
30,241 Ξ”
jax-v0.4.23 Released December 14, 2023
0 Ξ”
jax-v0.4.23-rc Released December 14, 2023
930 Ξ”
jax-v0.4.22 Released December 13, 2023
0 Ξ”
jax-v0.4.22-rc2 Released December 13, 2023
132 Ξ”
jax-v0.4.22-rc Released December 13, 2023
7,434 Ξ”
jax-v0.4.21 Released December 4, 2023
0 Ξ”
jax-v0.4.21-rc Released December 4, 2023
15,642 Ξ”
jax-v0.4.20 Released November 2, 2023
0 Ξ”
jax-v0.4.20-rc Released November 2, 2023
5,123 Ξ”

Top Contributors in jax-v0.4.22-rc

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

Directory Browser for jax-v0.4.22-rc

All files are compared to previous version, jax-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