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

Jax: jax-v0.4.21-rc Release

Release date:
December 4, 2023
Previous version:
jax-v0.4.20 (released November 2, 2023)
Magnitude:
15,642 Diff Delta
Contributors:
32 total committers
Data confidence:
Commits:
267 impacting commits

267 Commits in this Release

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

[shape_poly] Copy many of the jax2tf/shape_poly_test to live outside of jax2tf.
Authored November 19, 2023
Use MLIR generated convenience functions athing(...) instead of writing AThingOp(...).result.
Authored November 17, 2023
[Mosaic] Python bindings for `VectorLayout`, `VRegDataBounds`, `assemble`, `disassemble`, `relayout` and `apply_layout_op`
Authored November 21, 2023
JAX tutorials: pseudorandom numbers
Authored November 21, 2023
tests: improve warnings-related tests
Authored November 30, 2023
[Mosaic] Expose C API for VectorLayout, VRegDataBounds
Authored November 4, 2023
Remove PythonJitTest from api_test.py.
Authored November 13, 2023
[shape_poly] Simplify the API for processing polymorphic_shape specifications
Authored November 23, 2023
Add a lax.platform_dependent API for writing platform-dependent code.
Authored October 27, 2023
[export] Simplify the handling of shardings in Exported.
Authored December 1, 2023
[Pallas] Enable interpreter mode as default lowering for CPU
Authored November 9, 2023
[Mosaic][NFC] C++ apply_vector_layout: Do not pass `RewriteContext` to functions that only need `target_shape`
Authored November 8, 2023
[Pallas] Refactor Mosaic lowering to encapsulate jaxpr->mlir type creation all in one place
Authored November 18, 2023
Replace apply_primitive internals with `jax.jit`.
Authored November 30, 2023
JAX tutorials: add automatic vectorization
Authored November 21, 2023
Remove nightly NVIDIA GPU multiprocess CI.
Authored December 1, 2023
[shard_map] implement eager custom_jvp / custom_vjp
Authored November 20, 2023
[PJRT C API] Change build wheel script to build a separate package for cuda kernels.
Authored November 6, 2023
[shape_poly] Simplify the indexing with slice to make it compatible with shape polymorphism
Authored November 27, 2023
Cleanup the code to picking lowering rules based on platform.
Authored November 3, 2023
Automate arguments for jax.distributed.initialize for cloud TPU environments.
Authored December 1, 2023
[export] Add and fix a test for exporting higher-order gradients with sharding
Authored December 4, 2023
Default-enable the Jax persistent compilation cache.
Authored November 27, 2023
[shape_poly] Refactor shape_poly_test in preparation for moving out of jax2tf.
Authored November 12, 2023
[XLA:Mosaic] Support arbitrary aligned shape for tpu.bitcast and support bitcast with bitwidth change in element.
Authored November 15, 2023
Reverts a80d35f4eda99bac42a1573ff5029e7035ebe71b
Authored November 10, 2023
[shard_map] add eager axis_index implementation and tests
Authored November 19, 2023
Replace gcc with clang compiler in CI nightly jobs.
Authored November 10, 2023
[shard-map] fix transpose replication checking bug with integer_pow
Authored November 28, 2023
jax.vmap: improve docs & error for structured in_axes
Authored November 15, 2023
Typecheck avals and sharding for arguments that were DCE'd.
Authored November 28, 2023
Add minimum version checks for cublas and cusparse.
Authored November 18, 2023
Integrate LLVM at llvm/llvm-project@9bdbb8226e70
Authored November 20, 2023
[shard-map] replace jaxpr interpreters with final-style-xform-of-eval-jaxpr
Authored November 29, 2023
[Mosaic] Add support for extracting the first element of a vector as a scalar
Authored November 7, 2023
improve psum_scatter docstring (formatting and content)
Authored November 15, 2023
[Pallas/Mosaic] Add support for barrier semaphores
Authored November 29, 2023
Reverts f0382a5838f4526d21631e804f6fe576bfc3f97e
Authored December 2, 2023
Ensure sharding-related array properties are documented
Authored November 3, 2023
[array-api] add simple smoketest target for standard CI testing
Authored November 27, 2023

Browse Other Releases

Latest Pending Unreleased 😎
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 Ξ”
jax-v0.4.19 Released October 18, 2023
0 Ξ”
jax-v0.4.19-rc Released October 18, 2023
5,722 Ξ”

Top Contributors in jax-v0.4.21-rc

gnecula
jakevdp
hawkinsp
tlongeri
mattjj
yashk2810
a-googler
sharadmv
jyingl3
apaszke

Directory Browser for jax-v0.4.21-rc

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