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

Jax: jaxlib-v0.4.21 Release

Release date:
December 4, 2023
Previous version:
jaxlib-v0.4.20 (released November 2, 2023)
Magnitude:
18,541 Diff Delta
Contributors:
31 total committers
Data confidence:
Commits:
271 impacting commits

271 Commits in this Release

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

Increase minimum jaxlib version to 0.4.19
Authored November 17, 2023
Add jax.experimental.array_api interface
Authored November 16, 2023
[shape_poly] Copy many of the jax2tf/shape_poly_test to live outside of jax2tf
Authored November 19, 2023
Add a private API to allow setting layouts on jitted computations
Authored November 15, 2023
Use MLIR generated convenience functions athing(...) instead of writing AThingOp(...).result
Authored November 17, 2023
JAX tutorials: external callbacks
Authored November 21, 2023
[Mosaic] Python bindings for `VectorLayout`, `VRegDataBounds`, `assemble`, `disassemble`, `relayout` and `apply_layout_op`
Authored November 21, 2023
JAX tutorials: add intro to jit compilation
Authored November 21, 2023
[Mosaic] Expose C API for VectorLayout, VRegDataBounds
Authored November 4, 2023
JAX tutorials: pseudorandom numbers
Authored November 21, 2023
[Mosaic][NFC] C++ apply_vector_layout: Do not pass `RewriteContext` to functions that only need `target_shape`
Authored November 8, 2023
[export] Simplify the handling of shardings in Exported
Authored December 1, 2023
[Pallas] Refactor Mosaic lowering to encapsulate jaxpr->mlir type creation all in one place
Authored November 18, 2023
Deprecate the device() method of JAX arrays
Authored November 30, 2023
[shape_poly] Simplify the API for processing polymorphic_shape specifications
Authored November 23, 2023
Replace apply_primitive internals with `jax.jit`
Authored November 30, 2023
Remove PythonJitTest from api_test.py
Authored November 13, 2023
[PJRT C API] Change build wheel script to build a separate package for cuda kernels
Authored November 6, 2023
tests: improve warnings-related tests
Authored November 30, 2023
Remove nightly NVIDIA GPU multiprocess CI
Authored December 1, 2023
Add a lax.platform_dependent API for writing platform-dependent code
Authored October 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
JAX tutorials: add automatic vectorization
Authored November 21, 2023
Remove redundant jax2tf/tests/shape_poly_test tests
Authored November 28, 2023
[shape_poly] Simplify the indexing with slice to make it compatible with shape polymorphism
Authored November 27, 2023
Default-enable the Jax persistent compilation cache
Authored November 27, 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] implement eager custom_jvp / custom_vjp
Authored November 20, 2023
[Pallas] Enable interpreter mode as default lowering for CPU
Authored November 9, 2023
Replace gcc with clang compiler in CI nightly jobs
Authored November 10, 2023
Add minimum version checks for cublas and cusparse
Authored November 18, 2023
jax.vmap: improve docs & error for structured in_axes
Authored November 15, 2023
Integrate LLVM at llvm/llvm-project@9bdbb8226e70
Authored November 20, 2023
Typecheck avals and sharding for arguments that were DCE'd
Authored November 28, 2023
[shard-map] fix transpose replication checking bug with integer_pow
Authored November 28, 2023
[export] Add and fix a test for exporting higher-order gradients with sharding
Authored December 4, 2023
[Mosaic] Add support for extracting the first element of a vector as a scalar
Authored November 7, 2023
[shard-map] replace jaxpr interpreters with final-style-xform-of-eval-jaxpr
Authored November 29, 2023

Browse Other Releases

Latest Pending Unreleased 😎
jaxlib-v0.4.25 Released February 24, 2024
17,354 Ξ”
jaxlib-v0.4.24 Released February 5, 2024
30,896 Ξ”
jaxlib-v0.4.23 Released December 14, 2023
704 Ξ”
jaxlib-v0.4.22 Released December 13, 2023
6,291 Ξ”
jaxlib-v0.4.21 Released December 4, 2023
18,541 Ξ”
jaxlib-v0.4.20 Released November 2, 2023
4,475 Ξ”
jaxlib-v0.4.19 Released October 18, 2023
7,126 Ξ”
jaxlib-v0.4.18 Released October 6, 2023
4,220 Ξ”
jaxlib-v0.4.17 Released October 3, 2023
9,022 Ξ”
jaxlib-v0.4.16 Released September 18, 2023
9,933 Ξ”

Top Contributors in jaxlib-v0.4.21

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

Directory Browser for jaxlib-v0.4.21

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