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

Jax: jax-v0.4.25-rc Release

Release date:
February 24, 2024
Previous version:
jax-v0.4.24 (released February 5, 2024)
Magnitude:
19,369 Diff Delta
Contributors:
39 total committers
Data confidence:
Commits:
227 impacting commits

227 Commits in this Release

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

Added ir.Value-based versions of some triton.compat APIs
Authored February 13, 2024
[shape_poly] Add limited support for equality explicit constraints
Authored January 27, 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
[key reuse] remove alternate simple implementation
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
[Mosaic] In apply_vector_layout.cc, check layout validity when reading the attribute
Authored February 15, 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
add unit test and move to _src/cudnn dir
Authored January 2, 2024
[attrs] add a jvp function with attrs support
Authored February 3, 2024
fix flash attention && example for vmap inference
Authored December 7, 2023
add custom parition & vmap
Authored December 4, 2023
add flash attention and different patterns
Authored November 21, 2023
[shape_poly] Change the representation of symbolic dimension expressions
Authored February 3, 2024
[XLA:Mosaic] Expose tpu::RotateOp with stride
Authored February 13, 2024
Removes unused IREE config parameters from tests
Authored February 13, 2024
Add jax.tree module with aliases of jax.tree_util
Authored February 12, 2024
Pallas Add custom pipeline scalar + None block shape dim support + while TPU lowering rule
Authored February 13, 2024
Dropped the Triton* prefix from class names in triton/lowering.py
Authored February 13, 2024
add jax.cudnn & add check for bias/mask sharding
Authored January 30, 2024
Split name_stack out of mlir.ModuleContext
Authored February 20, 2024
jax.numpy: implement scalar boolean indexing
Authored February 8, 2024
Add Pallas attention kernel for GPU serving
Authored February 15, 2024
[shmap] Support multiple axes for standard collectives in shard_map
Authored January 31, 2024
Reverts 8e96c49f85321a4df9549a9f3f813bb7eb71b139
Authored February 13, 2024
Reverts b506fee9e389391efb1336bc7575dba913e75cdf
Authored February 19, 2024
address some format issues
Authored January 18, 2024
add an example
Authored December 4, 2023
Tighten up typing of optional config values
Authored February 12, 2024
init
Authored November 6, 2023
[attrs] allow passing a jax-attrs object to jit functions
Authored February 14, 2024
Don't call inspect.signature() each time we trace a jit()
Authored February 15, 2024
[PJRT C API] Supports plugins to register a method to create topology
Authored February 23, 2024
Define reuse_key primitive in jax._src.prng
Authored February 14, 2024
[XLA:Mosaic] Unify ext/trunc in infer vector layout
Authored February 23, 2024
rename tests with more descriptive name & Unify SDPA API
Authored January 19, 2024
Support activation offloading to host in JAX!
Authored February 6, 2024
Use ir.FloatType instead of a Pallas-local shim
Authored February 16, 2024

Browse Other Releases

Latest Pending Unreleased 😎
jax-v0.4.27-rc Released May 7, 2024
37,017 Ξ”
jax-v0.4.26 Released April 3, 2024
0 Ξ”
jax-v0.4.26-rc Released May 18, 2024
19,635 Ξ”
jax-v0.4.25 Released February 24, 2024
0 Ξ”
jax-v0.4.25-rc Released February 24, 2024
19,369 Ξ”
jax-v0.4.24 Released February 5, 2024
0 Ξ”
jax-v0.4.24-rc Released February 5, 2024
35,489 Ξ”
jax-v0.4.23 Released December 14, 2023
0 Ξ”
jax-v0.4.23-rc Released December 14, 2023
930 Ξ”
jax-v0.4.22 Released December 13, 2023
0 Ξ”

Top Contributors in jax-v0.4.25-rc

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

Directory Browser for jax-v0.4.25-rc

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