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

Jax: jaxlib-v0.4.32 Release

Release date:
September 11, 2024
Previous version:
jaxlib-v0.4.31 (released July 29, 2024)
Magnitude:
23,380 Diff Delta
Contributors:
58 total committers
Data confidence:
Commits:
373 impacting commits

373 Commits in this Release

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

refactor jax.lax to not depend on jax.numpy
Authored September 1, 2024
First pass at ufunc interfaces for several jax.numpy functions
Authored August 30, 2024
Clean up BUILD files
Authored August 26, 2024
Partial rollback of https://github.com/google/jax/pull/23353 as discussed in https://github.com/google/jax/pull/23353#issuecomment-2326604708
Authored September 3, 2024
Implement initial vmap over pallas_call w/ ragged inputs (via jumbles)
Authored August 16, 2024
document jax.Array methods and attributes
Authored August 16, 2024
[Pallas][XLA:Mosaic] Add python stack traces to Mosaic errors that occur in Pallas
Authored August 12, 2024
Fix a number of minor problems in the ROCM build
Authored August 27, 2024
Implement symmetric_product() to produce a symmetric matrix: `C = alpha * X @ X.T + beta * C`
Authored September 6, 2024
Initial integration of sharding in types in JAX. 0
Authored August 29, 2024
[Pallas] Implement tiled and swizzled Memref loads for Mosaic GPU via "GPUBlockSpec"
Authored September 11, 2024
Port QR factorization GPU kernel to FFI
Authored August 20, 2024
Standardize default layout to `None` in internals (dispatch, lowering and compilation) and non-default layouts to concrete layouts
Authored August 28, 2024
feat(lib): add real-valued implementation of jax.scipy.special.fresnel
Authored September 3, 2024
Activate QR Factorization to XLA's FFI
Authored August 23, 2024
[shape_polyO] Improve handling of equality shape constraints
Authored September 6, 2024
Support variable sequence lengths
Authored July 18, 2024
Introduce hermetic CUDA in Google ML projects
Authored August 14, 2024
PR #23223: [NVIDIA] Reduce number of tests for `jax.nn.dot_product_attention`
Authored August 30, 2024
Reverts 66a3f87a24016594794c2ee289826baed5e979a4
Authored August 20, 2024
Added basic pipelining to Pallas on Mosaic GPU
Authored September 3, 2024
Add cudnn_fusion decorator lowering computations to XLA cuDNN fusions
Authored July 24, 2024
Add support for the DeviceLocalLayout API when lowering FFI calls
Authored September 5, 2024
Port the GPU Cholesky update custom call to the FFI
Authored August 20, 2024
Register several jax.numpy argument name deprecations
Authored August 22, 2024
Skip the global jit cpp cache if in/out_layouts are not None
Authored August 20, 2024
refactor lax.loops to avoid importing from jax.numpy
Authored August 28, 2024
jnp.setxor1d: add support for static size argument
Authored August 13, 2024
Port GPU kernel for symmetric eigendecomposition to GPU
Authored September 6, 2024
[Pallas] Add run_scoped interpret mode rule + Enable DMA tests
Authored August 15, 2024
Determine LAPACK workspaces during SVD kernel runtime
Authored August 13, 2024
Improve docs for jnp.split & related APIs
Authored September 9, 2024
Add TritonCompilerParams for specifying compiler arguments instead of a dict
Authored September 4, 2024
Refactor & document cumulative reductions
Authored September 3, 2024
Fixed some type errors under pyright
Authored September 4, 2024
add sliding window attn
Authored August 30, 2024
Added pl.CompilerParams subclass for Mosaic GPU
Authored September 4, 2024
Support strided load / store in interpret mode
Authored August 28, 2024
Create `jax.make_mesh(axis_shapes: Sequence[int], axis_names: Sequence[str], devices: Sequence[jax.Device] | None = None)` API to make it easier to create a mesh and reduce a ton of boilerplate
Authored September 3, 2024
[Mosaic GPU] Add support for short n dimension in WGMMA
Authored August 23, 2024

Browse Other Releases

Latest Pending Unreleased 😎
0.3.23 Released October 12, 2022
0 Ξ”
0.3.22 Released October 11, 2022
22 Ξ”
0.3.21 Released September 30, 2022
0 Ξ”
0.3.20 Released September 28, 2022
0 Ξ”
jaxlib-v0.4.32 Released September 11, 2024
23,380 Ξ”
jaxlib-v0.4.31 Released July 29, 2024
41,391 Ξ”
jaxlib-v0.4.30 Released June 18, 2024
9,820 Ξ”
jaxlib-v0.4.29 Released June 10, 2024
27,516 Ξ”
jaxlib-v0.4.28 Released May 9, 2024
0 Ξ”
jaxlib-v0.4.27 Released May 7, 2024
17,911 Ξ”

Top Contributors in jaxlib-v0.4.32

jakevdp
a-googler
yashk2810
justinjfu
hawkinsp
superbobry
dfm
apaszke
rajasekharporeddy
pparuzel

Directory Browser for jaxlib-v0.4.32

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

Release Notes Published

WARNING: This release has been yanked from PyPI because of a data corruption bug on TPU if there are multiple TPU slices in the job

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