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

Jax: jax-v0.4.27-rc Release

Release date:
May 7, 2024
Previous version:
jax-v0.4.26 (released April 3, 2024)
Magnitude:
18,638 Diff Delta
Contributors:
57 total committers
Data confidence:
Commits:
347 impacting commits

347 Commits in this Release

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

Add `Layout` support to `jax.jit`.
Authored April 6, 2024
DOC: add introduction to sharded computation
Authored April 16, 2024
[Mosaic GPU] Use a custom TMA descriptor initialization method
Authored April 26, 2024
[Mosaic GPU] Add the flash attention example
Authored April 29, 2024
[Mosaic] Expand support of vector.extract and vector.extract_strided_slice
Authored May 1, 2024
DOC: one last readthrough of the new 101 tutorials
Authored April 17, 2024
Add discharge rules for scan with mutable arrays. Move mutable array tests to separate file.
Authored April 29, 2024
Workaround numpy 1.x assert_allclose false-positive result in comparing complex infinities.
Authored April 4, 2024
DOC: Improve remaining jax.scipy docstrings
Authored May 2, 2024
Revert "Doc: Fix example code snippet format in jax.experimental.multihost_utils.global_array_to_host_local_array doc"
Authored April 12, 2024
Bumped the minimum jaxlib to 0.4.23
Authored April 4, 2024
DOC: improve documentation of where, nonzero, and related functions
Authored April 26, 2024
Refactor QDWH to be more efficient when run batched under vmap.
Authored April 25, 2024
[Pallas TPU] Generalize while_loop lowering in Pallas -> Mosaic.
Authored April 18, 2024
[export] Add backwards compatibility test for Pallas call on GPUs.
Authored May 2, 2024
Make `core.Token` a non-trivial class which wraps a `jax.Array`. Currently, we use a singleton and empty `core.token` object everywhere. After the change, tokens could be created and threaded in and out of computations to build up dependency.
Authored April 18, 2024
Add support for max_version, dl_device, copy kwargs in __dlpack__
Authored April 11, 2024
Compute source maps when pretty-printing jaxprs.
Authored May 6, 2024
Add JAX API that provides sparse matmul support (2:4 structured sparsity)
Authored April 24, 2024
tweak jit docstring
Authored April 20, 2024
[Mosaic] Support scf.while and scf.condition.
Authored April 16, 2024
jnp.linalg tensorinv & tensorsolve: improve implementation & docs
Authored May 6, 2024
Add __array_namespace_info__ and corresponding utilities
Authored April 11, 2024
Refactored common upcast for integral-type accumulators
Authored May 6, 2024
Remove explicit pallas trace_start/trace_stop primitives. These are now automatically inserted with the usage of jax.named_scope.
Authored April 27, 2024
Remove last scipy imports
Authored May 3, 2024
Update `from_dlpack` to match array API 2023
Authored April 4, 2024
Jax persistent compilation cache user guide.
Authored April 9, 2024
Allow multiple indexers when doing discharge or swap in pallas
Authored May 1, 2024
[Mosaic GPU] Implement a simple profilng tool using CUDA events
Authored April 25, 2024
Added a GPU-specific approximate tanh to Pallas
Authored April 26, 2024
DOC: improve docs of transpose & matrix_transpose
Authored April 26, 2024
If callback returns a fully replicated global array, return it as is.
Authored April 14, 2024
reintroduce the Threefry GPU kernel lowering, under a flag
Authored May 1, 2024
[Pallas][Mosaic] Expose semaphore read.
Authored April 10, 2024
Finalize the deprecation of the arr.device() method
Authored April 9, 2024
Expose existing functions in array API namespace
Authored April 15, 2024
Optimize `jax.device_put()` dispatch for 1:1 device-to-device transfers
Authored April 22, 2024
Add layout support to `make_array_from_callback`.
Authored April 15, 2024
Restore upstream-nightly github action
Authored April 4, 2024

Browse Other Releases

Latest Pending Unreleased 😎
jax-v0.4.29 Released June 10, 2024
0 Ξ”
jax-v0.4.29-rc Released June 10, 2024
27,592 Ξ”
jax-v0.4.28 Released May 9, 2024
0 Ξ”
jax-v0.4.28-rc Released May 9, 2024
0 Ξ”
jax-v0.4.27 Released May 7, 2024
0 Ξ”
jax-v0.4.27-rc Released May 7, 2024
18,638 Ξ”
jax-v0.4.26 Released April 3, 2024
0 Ξ”
jax-v0.4.26-rc Released April 3, 2024
17,079 Ξ”
jax-v0.4.25 Released February 24, 2024
0 Ξ”
jax-v0.4.25-rc Released February 24, 2024
17,651 Ξ”

Top Contributors in jax-v0.4.27-rc

jakevdp
yashk2810
a-googler
Micky774
apaszke
superbobry
mattjj
pearu
rajasekharporeddy
justinjfu

Directory Browser for jax-v0.4.27-rc

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