Home » Oreps » Jax » All Releases » jax-v0.4.37 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.37
Basic Details Issues Addressed Top Contributors Directory Browser Release Notes
Jax jax-v0.4.37

Jax: jax-v0.4.37 Release

Release date:
December 12, 2024
Previous version:
jax-v0.4.36 (released January 27, 2025)
Magnitude:
4,817 Diff Delta
Contributors:
22 total committers
Data confidence:
Commits:
43 impacting commits

43 Commits in this Release

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

A test for using symbolic shapes for minformer
Authored December 6, 2024
[Mosaic:TPU] For trunc, expand supported tilings and offsets
Authored December 9, 2024
Refactor array serialization into separate JAX and tensorstore logic
Authored December 6, 2024
Add lax.composite primitive
Authored November 21, 2024
Activate Tridiagonal Reduction to XLA's FFI
Authored September 4, 2024
[pallas:mosaic_gpu] Added a lowering rule for the general `lax.while_loop_p`
Authored December 6, 2024
Remove dead code after minimum jaxlib version bump to v0.4.36
Authored December 6, 2024
[pallas:triton] Add support for `DotAlgorithmPreset` `precision` arguments to `dot`
Authored November 28, 2024
[Mosaic TPU] Allow downgrading the IR during serialization for forward compat
Authored December 3, 2024
Activate Triangular Solve to XLA's FFI
Authored July 17, 2024
[Pallas TPU] Improve lowerings for boolean comparison operations
Authored December 3, 2024
Partial discharge for scan_p ops
Authored September 30, 2024
[Pallas] Add support for run_state to cost estimator
Authored December 6, 2024
Activate Schur Decomposition to XLA's FFI
Authored December 6, 2024
Simplify implementation of nn.relu. Update paper link
Authored December 8, 2024
Activate Tridiagonal Reduction to XLA's FFI
Authored December 9, 2024
[pallas] Argument reordering without jaxpr_eval during scan_p discharge
Authored December 6, 2024
[export] Improved the documentation
Authored December 8, 2024
[JAX] Add concurrent execution support in colocated Python
Authored December 5, 2024
Use private names for args in api_util to avoid shadowing kwargs keys
Authored December 9, 2024
[pallas:mosaic_gpu] `FragmentedArray.reduce_sum` now returns a `FragmentedArray`
Authored December 6, 2024
[Pallas MGPU] Use multiple k/v_consumed_barriers in the attention kernel
Authored December 6, 2024
Use boolean values for partial mask blocks in the splash attention kernel
Authored December 4, 2024
Avoid index out of range error in carry structure check
Authored December 9, 2024
[Pallas MGPU] Disable XLA:GPU autotuning in attention tests
Authored December 6, 2024
[pallas] Add `DotAlgorithmPreset` note to CHANGELOG
Authored December 6, 2024
Set -Werror=mismatched-tags on clang
Authored December 3, 2024
Update versions for v0.4.37 release
Authored December 9, 2024
Bump JAX version after release
Authored December 6, 2024
[mosaic_gpu] Emit a slightly more informative error message in `FragmentedArray._pointwise`
Authored December 6, 2024
Support transfer guard in broadcast_one_to_all(). 0
Authored December 7, 2024
Finish 0.4.36 release
Authored December 6, 2024
Fix type annotation for numpy.linalg.matrix_norm argument 'ord'
Authored December 9, 2024
Add a flag to enable detailed timestamped logging of subprocess commands
Authored December 4, 2024
[pallas:triton] Add support for `DotAlgorithmPreset` `precision` arguments to `dot`
Authored December 9, 2024
Update XLA dependency to use revision
Authored December 6, 2024
[Mosaic TPU] Allow downgrading the IR during serialization for forward compat
Authored December 9, 2024
Temporarily allow bfloat16 dot algorithms on CPU
Authored December 7, 2024
Set -Werror=mismatched-tags on clang
Authored December 6, 2024
Add a flag to enable detailed timestamped logging of subprocess commands
Authored December 6, 2024

Browse Other Releases

Latest Pending Unreleased 😎
jax-v0.5.3 Released March 31, 2025
45,270 Ξ”
jax-v0.5.2 Released March 4, 2025
21 Ξ”
jax-v0.5.1 Released April 5, 2025
73,856 Ξ”
jax-v0.5.0 Released February 15, 2025
43,803 Ξ”
jax-v0.4.38 Released January 16, 2025
17,793 Ξ”
jax-v0.4.37 Released December 12, 2024
4,817 Ξ”
jax-v0.4.36 Released January 27, 2025
85,862 Ξ”
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 Ξ”

Top Contributors in jax-v0.4.37

gnecula
pparuzel
tlongeri
rdyro
ghpvnist
superbobry
hawkinsp
apaszke
cperivol
chr1sj0nes

Directory Browser for jax-v0.4.37

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

Release Notes Published

This is a patch release of jax 0.4.36. Only "jax" was released at this version.

  • Bug fixes
    • Fixed a bug where jit would error if an argument was named f (#25329).
    • Fix a bug that will throw index out of range error in jax.lax.while_loop if the user registers pytree node class with different aux data for the flatten and flatten_with_path.
    • Pinned a new libtpu release (0.0.6) that fixes a compiler bug on TPU v6e.