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

Jax: jaxlib-v0.4.27 Release

Release date:
May 7, 2024
Previous version:
jaxlib-v0.4.26 (released May 18, 2024)
Magnitude:
36,331 Diff Delta
Contributors:
57 total committers
Data confidence:
Commits:
343 impacting commits

343 Commits in this Release

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

DOC: Improve docstrings for jax.scipy.linalg
Authored May 2, 2024
DOC: update installation guide
Authored April 16, 2024
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. 0
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
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`
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
Refactored common upcast for integral-type accumulators
Authored May 6, 2024
Remove explicit pallas trace_start/trace_stop primitives. 0
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 😎
jaxlib-v0.4.31 Released July 29, 2024
49,788 Ξ”
jaxlib-v0.4.30 Released June 18, 2024
13,835 Ξ”
jaxlib-v0.4.29 Released June 10, 2024
30,491 Ξ”
jaxlib-v0.4.28 Released May 9, 2024
0 Ξ”
jaxlib-v0.4.27 Released May 7, 2024
36,331 Ξ”
jaxlib-v0.4.26 Released May 18, 2024
19,420 Ξ”
jaxlib-v0.4.25 Released February 24, 2024
19,014 Ξ”
jaxlib-v0.4.24 Released February 5, 2024
35,114 Ξ”
jaxlib-v0.4.23 Released December 14, 2023
930 Ξ”
jaxlib-v0.4.22 Released December 13, 2023
7,552 Ξ”

Top Contributors in jaxlib-v0.4.27

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

Directory Browser for jaxlib-v0.4.27

We haven't yet finished calculating and confirming the files and directories changed in this release. Please check back soon.