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:
3,574 Diff Delta
Contributors:
22 total committers
Data confidence:
Commits:
42 impacting commits

42 Commits in this Release

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

[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
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 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:mosaic_gpu] `FragmentedArray.reduce_sum` now returns a `FragmentedArray`
Authored December 6, 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(). Fixes https://github.com/jax-ml/jax/issues/25325
Authored December 7, 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
Finish 0.4.36 release
Authored December 6, 2024
[pallas] Add `DotAlgorithmPreset` note to CHANGELOG.
Authored December 9, 2024
[pallas:triton] Add support for `DotAlgorithmPreset` `precision` arguments to `dot`.
Authored December 9, 2024
[Mosaic TPU] Allow downgrading the IR during serialization for forward compat
Authored December 9, 2024
[Pallas TPU] Improve lowerings for boolean comparison operations
Authored December 3, 2024
Update XLA dependency to use revision
Authored December 6, 2024
Update XLA dependency to use revision
Authored December 7, 2024
Temporarily allow bfloat16 dot algorithms on CPU.
Authored December 7, 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 19, 2025
42,198 Ξ”
jax-v0.5.2 Released March 4, 2025
21 Ξ”
jax-v0.5.1 Released February 26, 2025
60,641 Ξ”
jax-v0.5.0 Released February 15, 2025
41,443 Ξ”
jax-v0.4.38 Released January 16, 2025
15,869 Ξ”
jax-v0.4.37 Released December 12, 2024
3,574 Ξ”
jax-v0.4.36 Released January 27, 2025
71,790 Ξ”
jax-v0.4.35 Released October 22, 2024
0 Ξ”
jax-v0.4.34 Released October 2, 2024
26,451 Ξ”
jax-v0.4.33 Released September 16, 2024
0 Ξ”

Top Contributors in jax-v0.4.37

pparuzel
tlongeri
rdyro
ghpvnist
superbobry
hawkinsp
apaszke
cperivol
chr1sj0nes
justinjfu

Directory Browser for jax-v0.4.37

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

Loading File Browser...

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.
GitClear uses cookies to ensure you get the best experience on our website. Learn more