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

Jax: jaxlib-v0.4.25 Release

Release date:
February 24, 2024
Previous version:
jaxlib-v0.4.24 (released February 5, 2024)
Magnitude:
17,354 Diff Delta
Contributors:
39 total committers
Data confidence:
Commits:
226 impacting commits

226 Commits in this Release

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

Changed lower_jaxpr_to_triton_ir to use ir.Value instead of triton.compat APIs
Authored February 13, 2024
Added ir.Value-based versions of some triton.compat APIs
Authored February 13, 2024
[shape_poly] Cleaning up naming of terms and factors
Authored February 20, 2024
[shape_poly] Performance improvements for symbolic dimension manipulations (step 3)
Authored February 15, 2024
[shape_poly] Add limited support for equality explicit constraints
Authored January 27, 2024
[key reuse] remove alternate simple implementation
Authored February 15, 2024
[Mosaic] In apply_vector_layout.cc, check layout validity when reading the attribute
Authored February 15, 2024
[JAX] Add an option subset_by_index that allows computing a contiguous subset of singular components from svd
Authored February 16, 2024
Refactor and optimize the implementation of config options
Authored February 12, 2024
[shape_polyO] Performance improvements for symbolic dimension manipulations (step 2)
Authored February 14, 2024
Inlined triton.compat
Authored February 14, 2024
fix flash attention && example for vmap inference
Authored December 7, 2023
add custom parition & vmap
Authored December 4, 2023
add unit test and move to _src/cudnn dir
Authored January 2, 2024
[XLA:Mosaic] Expose tpu::RotateOp with stride
Authored February 13, 2024
[shape_poly] Change the representation of symbolic dimension expressions
Authored February 3, 2024
Add jax.tree module with aliases of jax.tree_util
Authored February 12, 2024
[attrs] add a jvp function with attrs support
Authored February 3, 2024
Dropped the Triton* prefix from class names in triton/lowering.py
Authored February 13, 2024
[attrs] add linearize and vjp support
Authored February 22, 2024
replace pjit with jit and only allow shardings on batch/head dim
Authored January 18, 2024
Split name_stack out of mlir.ModuleContext
Authored February 20, 2024
Pallas Add custom pipeline scalar + None block shape dim support + while TPU lowering rule
Authored February 13, 2024
Removes unused IREE config parameters from tests
Authored February 13, 2024
jax.numpy: implement scalar boolean indexing
Authored February 8, 2024
Reverts 8e96c49f85321a4df9549a9f3f813bb7eb71b139
Authored February 13, 2024
Tighten up typing of optional config values
Authored February 12, 2024
init
Authored November 6, 2023
address some format issues
Authored January 18, 2024
add jax.cudnn & add check for bias/mask sharding
Authored January 30, 2024
[PJRT C API] Supports plugins to register a method to create topology
Authored February 23, 2024
[XLA:Mosaic] Unify ext/trunc in infer vector layout
Authored February 23, 2024
[attrs] allow passing a jax-attrs object to jit functions
Authored February 14, 2024
Add Pallas attention kernel for GPU serving
Authored February 15, 2024
Reverts b506fee9e389391efb1336bc7575dba913e75cdf
Authored February 19, 2024
[shmap] Support multiple axes for standard collectives in shard_map
Authored January 31, 2024
Define reuse_key primitive in jax._src.prng
Authored February 14, 2024
Use ir.FloatType instead of a Pallas-local shim
Authored February 16, 2024
Reland https://github.com/google/jax/pull/10573
Authored February 16, 2024
add an example
Authored December 4, 2023

Browse Other Releases

Latest Pending Unreleased 😎
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 Ξ”
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 Ξ”
jaxlib-v0.4.21 Released December 4, 2023
18,528 Ξ”

Top Contributors in jaxlib-v0.4.25

superbobry
gnecula
Cjkkkk
hawkinsp
jakevdp
a-googler
bythew3i
mattjj
tlongeri
yashk2810

Directory Browser for jaxlib-v0.4.25

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