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

Jax: jaxlib-v0.4.31 Release

Release date:
July 29, 2024
Previous version:
jaxlib-v0.4.30 (released June 18, 2024)
Magnitude:
43,313 Diff Delta
Contributors:
66 total committers
Data confidence:
Commits:
531 impacting commits

531 Commits in this Release

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

Delete `xmap` and the `jax.experimental.maps` module. 0
Authored July 24, 2024
[pallas] Break long lines in the Pallas docs
Authored June 25, 2024
Run `pyupgrade --py310-plus`
Authored June 26, 2024
Reverts 0d058ce86f04a44a51abba1261768fb46edf69d9
Authored July 25, 2024
[pallas] More simplification of grid mapping and calling convention
Authored July 23, 2024
[Mosaic GPU] Move barrier allocation to SMEM scratch specs
Authored July 24, 2024
Bump minimum jaxlib version to v0.4.30
Authored June 18, 2024
[pallas] Added more documentation for grid and BlockSpec
Authored June 27, 2024
remove named_shapes (since xmap is now gone)
Authored July 25, 2024
Fix C++ registration of FFI handlers and consolidate gpu/linalg kernel implementation
Authored July 10, 2024
[ROCm] Bring-up pjrt support
Authored June 12, 2024
[export] Add documentation for debugging and for ensuring compatibility
Authored June 19, 2024
[JAX] Add caching to pjit._infer_params
Authored June 21, 2024
Add support to pipeline emitter for shapes that don't perfectly divide the block shapes
Authored June 18, 2024
Add concrete layout API to JAX. 0
Authored June 27, 2024
[Pallas] Implement block-invariant sampling
Authored June 24, 2024
[Mosaic GPU] Add a warp specialized kernel with a separate TMA warp
Authored June 28, 2024
[pallas] Improve some error messages and add API tests
Authored July 2, 2024
Another take at enabling Pallas GPU tests on x64
Authored July 22, 2024
Internal: remove binary operation ufunc factories
Authored June 24, 2024
Documented a few more Pallas APIs and added them to the API docs
Authored July 19, 2024
Internal: remove one_to_one_unop factory
Authored June 24, 2024
PR #21371: [NVIDIA] Add new SDPA API to jax.nn
Authored July 8, 2024
Changed the naming of internal config APIs
Authored April 15, 2024
Added basic support for int2/uint2 dtypes to JAX
Authored July 4, 2024
[pallas] Move some tests from tpu/pallas_call_test to pallas_test
Authored July 8, 2024
[pallas] Simplify handling of BlockMapping and GridMapping
Authored July 21, 2024
[Mosaic GPU] Simplify the matmul example
Authored July 25, 2024
Clean up
Authored July 3, 2024
pjit.py cleanups
Authored June 20, 2024
_make_dispatch_table calls in Pallas GPU lowering are now slightly more compact
Authored June 18, 2024
Update persistent compilation cache doc
Authored July 4, 2024
[Pallas][Mosaic] Relax dynamic index on 2nd minor dim in load/store
Authored July 2, 2024
[pallas] Move tests to the GPU- and TPU- specific directories
Authored July 2, 2024
[Mosaic GPU] Use explicit WGMMA/ALU scheduling in the flash attention kernel
Authored June 20, 2024
Move FFI helper macros from jaxlib/cpu/lapack_kernels.cc to a jaxlib/ffi_helpers.h
Authored July 10, 2024
Port Householder Product to XLA's FFI
Authored July 12, 2024
Remove dead code now that xmap is deleted
Authored July 24, 2024
Migrated ``pl.BlockSpec`` uses in JAX to the new argument order
Authored July 2, 2024
Reverts 0e17d26b6d81a6b281f55bdd81a6f0ab45efeafe
Authored July 24, 2024

Browse Other Releases

Latest Pending Unreleased 😎
0.3.23 Released October 12, 2022
0 Ξ”
0.3.22 Released October 11, 2022
16 Ξ”
0.3.21 Released September 30, 2022
0 Ξ”
0.3.20 Released September 28, 2022
0 Ξ”
jaxlib-v0.4.32 Released September 11, 2024
22,228 Ξ”
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
0 Ξ”
jaxlib-v0.4.27 Released May 7, 2024
30,921 Ξ”

Top Contributors in jaxlib-v0.4.31

gnecula
yashk2810
apaszke
hawkinsp
superbobry
jakevdp
rajasekharporeddy
mattjj
sharadmv
dfm

Directory Browser for jaxlib-v0.4.31

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