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:
30,921 Diff Delta
Contributors:
57 total committers
Data confidence:
Commits:
356 impacting commits

356 Commits in this Release

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

DOC: replace old tutorials with new content
Authored April 18, 2024
jnp.linalg: improve API documentation
Authored May 6, 2024
Refactor array_api namespace, relying more directly on jax.numpy
Authored May 1, 2024
DOC: Improve docstrings for jax.scipy.linalg
Authored May 2, 2024
DOC: remove copy of thinking-in-jax from new tutorial flow
Authored April 17, 2024
DOC: add manual documentation to jax.scipy.special functions
Authored April 29, 2024
Avoid 'from jax import config' imports
Authored April 11, 2024
Initial commit for Mosaic GPU
Authored April 18, 2024
DOC: update installation guide
Authored April 16, 2024
Add `Layout` support to `jax.jit`
Authored April 6, 2024
[Mosaic GPU] Add the first example: pipelined matmul
Authored April 25, 2024
DOC: add introduction to sharded computation
Authored April 16, 2024
Switch Mosaic GPU to a custom pass pipeline and improve the lowering of GPU launch
Authored April 24, 2024
DOC: add stateful computation doc
Authored April 17, 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: Improve remaining jax.scipy docstrings
Authored May 2, 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
Bumped the minimum jaxlib to 0.4.23
Authored April 4, 2024
Add discharge rules for scan with mutable arrays. 0
Authored April 29, 2024
DOC: one last readthrough of the new 101 tutorials
Authored April 17, 2024
Add JAX API that provides sparse matmul support (2:4 structured sparsity)
Authored April 24, 2024
tweak jit docstring
Authored April 20, 2024
Add support for max_version, dl_device, copy kwargs in __dlpack__
Authored April 11, 2024
Pallas now exclusively uses XLA for compiling kernels on GPU
Authored April 4, 2024
[Mosaic] Support scf.while and scf.condition
Authored April 16, 2024
Workaround numpy 1.x assert_allclose false-positive result in comparing complex infinities
Authored April 4, 2024
[Pallas TPU] Generalize while_loop lowering in Pallas -> Mosaic
Authored April 18, 2024
Refactored common upcast for integral-type accumulators
Authored May 6, 2024
jnp.linalg tensorinv & tensorsolve: improve implementation & docs
Authored May 6, 2024
Remove last scipy imports
Authored May 3, 2024
Remove explicit pallas trace_start/trace_stop primitives. 0
Authored April 27, 2024
[export] Add backwards compatibility test for Pallas call on GPUs
Authored May 2, 2024
Compute source maps when pretty-printing jaxprs
Authored May 6, 2024
Make `core.Token` a non-trivial class which wraps a `jax.Array`
Authored April 18, 2024
Update `from_dlpack` to match array API 2023
Authored April 4, 2024
[Mosaic GPU] Implement a simple profilng tool using CUDA events
Authored April 25, 2024
Allow multiple indexers when doing discharge or swap in pallas
Authored May 1, 2024

Browse Other Releases

Latest Pending Unreleased 😎
jaxlib-v0.4.31 Released July 29, 2024
43,313 Ξ”
jaxlib-v0.4.30 Released June 18, 2024
11,389 Ξ”
jaxlib-v0.4.29 Released June 10, 2024
26,270 Ξ”
jaxlib-v0.4.28 Released May 9, 2024
5,052 Ξ”
jaxlib-v0.4.27 Released May 7, 2024
30,921 Ξ”
jaxlib-v0.4.26 Released May 18, 2024
16,029 Ξ”
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 Ξ”

Top Contributors in jaxlib-v0.4.27

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

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.

GitClear uses cookies to ensure you get the best experience on our website. Learn more about our cookie policy