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

Jax: jax-v0.4.32-rc Release

Release date:
September 11, 2024
Previous version:
jax-v0.4.31 (released July 29, 2024)
Magnitude:
35,570 Diff Delta
Contributors:
69 total committers
Data confidence:
Commits:
507 impacting commits

507 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
jnp.intersect1d: add support for static size argument
Authored August 10, 2024
[Pallas][XLA:Mosaic] Add python stack traces to Mosaic errors that occur in Pallas
Authored August 12, 2024
Add optional automatic remat optimization to custom_vjp
Authored July 29, 2024
Fix a number of minor problems in the ROCM build
Authored August 27, 2024
[Mosaic] Column shift relayouts for non-native tilings and packed types, except for (1, n) and packed
Authored August 9, 2024
[pallas] Improve the error localization
Authored July 18, 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
Bump minimum jaxlib version to 0.4.31. 0
Authored July 30, 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
Faster MHA backwards pass
Authored August 1, 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
Remove forward compatibility mode for old PRGN custom call on GPU
Authored July 31, 2024
Pallas Mosaic GPU lowering now supports (at least the basic) pl.BlockSpecs
Authored August 8, 2024
[NFC][Mosaic TPU] Start breaking up relayout into smaller pieces
Authored August 1, 2024
Support variable sequence lengths
Authored July 18, 2024
Introduce hermetic CUDA in Google ML projects
Authored August 14, 2024
Move jaxlib GPU handlers to separate build target
Authored August 1, 2024
Better docs for array, asarray, linspace
Authored August 9, 2024
[NFC][Mosaic TPU] Separate out retiling from relayout
Authored August 1, 2024
PR #23223: [NVIDIA] Reduce number of tests for `jax.nn.dot_product_attention`
Authored August 30, 2024
Reverts 66a3f87a24016594794c2ee289826baed5e979a4
Authored August 20, 2024
Init commit
Authored August 1, 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

Browse Other Releases

Latest Pending Unreleased 😎
jax-v0.4.35 Released October 22, 2024
0 Ξ”
jax-v0.4.34 Released October 2, 2024
29,039 Ξ”
jax-v0.4.33 Released September 16, 2024
0 Ξ”
jax-v0.4.33-rc Released September 16, 2024
43 Ξ”
jax-v0.4.32 Released September 11, 2024
0 Ξ”
jax-v0.4.32-rc Released September 11, 2024
35,570 Ξ”
jax-v0.4.31 Released July 29, 2024
0 Ξ”
jax-v0.4.31-rc Released July 29, 2024
51,898 Ξ”
jax-v0.4.30 Released June 18, 2024
0 Ξ”
jax-v0.4.30-rc Released June 18, 2024
13,835 Ξ”

Top Contributors in jax-v0.4.32-rc

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

Directory Browser for jax-v0.4.32-rc

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