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:
41,440 Diff Delta
Contributors:
66 total committers
Data confidence:
Commits:
522 impacting commits

522 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. It's been 5 months since its deprecation (more than the standard 3 months deprecation period).
Authored July 24, 2024
[pallas] Added more documentation for grid and BlockSpec.
Authored June 27, 2024
[export] Add documentation for debugging and for ensuring compatibility.
Authored June 19, 2024
[ROCm] Bring-up pjrt support
Authored June 12, 2024
[pallas] Move some tests from tpu/pallas_call_test to pallas_test
Authored July 8, 2024
[pallas] Improve some error messages and add API tests.
Authored July 2, 2024
Add support to pipeline emitter for shapes that don't perfectly divide the block shapes
Authored June 18, 2024
Fix C++ registration of FFI handlers and consolidate gpu/linalg kernel implementation.
Authored July 10, 2024
[Pallas] Implement block-invariant sampling.
Authored June 24, 2024
Add concrete layout API to JAX. The API takes `major_to_minor: tuple[int, ...]` and `tiling: tuple[tuple[int, ...], ...]` as the arguments. Allows users to pass layouts to `with_sharding_constraint` to constrain the layout + sharding.
Authored June 27, 2024
PR #21371: [NVIDIA] Add new SDPA API to jax.nn
Authored July 8, 2024
[JAX] Add caching to pjit._infer_params.
Authored June 21, 2024
Added basic support for int2/uint2 dtypes to JAX
Authored July 4, 2024
[pallas] Move tests to the GPU- and TPU- specific directories
Authored July 2, 2024
[Mosaic GPU] Add a warp specialized kernel with a separate TMA warp
Authored June 28, 2024
Update persistent compilation cache doc
Authored July 4, 2024
Internal: remove one_to_one_unop factory
Authored June 24, 2024
[Pallas] Add limited boolean memref support for scalars.
Authored July 16, 2024
Documented a few more Pallas APIs and added them to the API docs
Authored July 19, 2024
Shmallas, a.k.a. allow lowering shard_map + run_state to a pallas_call.
Authored July 23, 2024
Changed the naming of internal config APIs
Authored April 15, 2024
Clean up
Authored July 3, 2024
[Mosaic GPU] Simplify the matmul example
Authored July 25, 2024
[pallas] Document the indexing_mode=Unblocked()
Authored July 7, 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
Add `sharding` to `convert_element_type_p` primitive.
Authored July 9, 2024
Migrated ``pl.BlockSpec`` uses in JAX to the new argument order
Authored July 2, 2024
Squashed commit of the following:
Authored July 2, 2024
[Pallas][Mosaic] Relax dynamic index on 2nd minor dim in load/store.
Authored July 2, 2024
[jax2tf] Deprecate jax2tf with native_serialization=False or enable_xla=False.
Authored July 16, 2024
[Mosaic GPU] Use explicit WGMMA/ALU scheduling in the flash attention kernel
Authored June 20, 2024
Storing the last access time of a cache entry in a separate file
Authored June 17, 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
[pallas] Fix the interpreter for block_shape not dividing the overall shape
Authored July 1, 2024
Fix type annotations for jnp.poly* functions
Authored June 22, 2024
[pallas] Fix the handling of captured consts
Authored July 19, 2024
[Mosaic GPU] Pass in TMA descriptors through kernel parameters
Authored July 2, 2024

Browse Other Releases

Latest Pending Unreleased 😎
0.3.23 Released October 12, 2022
305 Ξ”
0.3.22 Released October 11, 2022
11,621 Ξ”
0.3.21 Released September 30, 2022
1,287 Ξ”
0.3.20 Released September 28, 2022
0 Ξ”
jaxlib-v0.4.31 Released July 29, 2024
41,440 Ξ”
jaxlib-v0.4.30 Released June 18, 2024
10,316 Ξ”
jaxlib-v0.4.29 Released June 10, 2024
27,453 Ξ”
jaxlib-v0.4.28 Released May 9, 2024
0 Ξ”
jaxlib-v0.4.27 Released May 7, 2024
17,951 Ξ”
jaxlib-v0.4.26 Released April 3, 2024
16,864 Ξ”

Top Contributors in jaxlib-v0.4.31

yashk2810
gnecula
apaszke
superbobry
jakevdp
rajasekharporeddy
hawkinsp
sharadmv
dfm
justinjfu

Directory Browser for jaxlib-v0.4.31

All files are compared to previous version, jaxlib-v0.4.30. Click here to browse diffs between other versions.

Loading File Browser...
GitClear uses cookies to ensure you get the best experience on our website. Learn more