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

Jax: jax-v0.4.24-rc Release

Release date:
February 5, 2024
Previous version:
jax-v0.4.23 (released December 14, 2023)
Magnitude:
30,241 Diff Delta
Contributors:
52 total committers
Data confidence:
Commits:
403 impacting commits

403 Commits in this Release

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

Upgrade JAX Advanced Autodiff 201
Authored December 15, 2023
[shard-map] add user tutorial
Authored January 8, 2024
[Pallas] Refactor indexing primitives to use NDIndexer abstraction
Authored January 2, 2024
[shape_poly] Add a decision procedure for inequalities.
Authored January 20, 2024
Pallas now uses MLIR Python builders to lower to Triton IR
Authored January 11, 2024
[shape_poly] Add more tests for reasoning about inequalities.
Authored January 24, 2024
[Pallas/Mosaic] Add support for nested `ref.at`
Authored January 3, 2024
[shape_poly] Fixes for the lexicographic ordering of monomials.
Authored January 5, 2024
[Mosaic] Remove Python implementation of apply_vector_layout and infer_memref_layout.
Authored January 10, 2024
Inlined some Triton-specific abstractions
Authored February 3, 2024
Inlined all builder.create_* calls
Authored February 5, 2024
jax.random: deprecate passing of batched keys to APIs
Authored January 17, 2024
add jax.explain_cache_misses tracing cache miss explanations
Authored June 9, 2023
Remove indices and devices from shard_arg_handlers and shard_args.
Authored January 5, 2024
Generate Python bindings for the Triton MLIR dialect
Authored January 2, 2024
Cache ClosedJaxpr creation to minimize cache_misses. ClosedJaxpr should always be created under a cache.
Authored December 22, 2023
Added a compatibility overlay for Triton Python APIs
Authored January 9, 2024
[XLA:Mosaic] Support inputs and outputs in scf::ForOp and add tpu::AssumeLayoutOp to work around block argument as operand.
Authored January 10, 2024
[Pallas][Mosaic] Add support for nontrivial semaphore memrefs
Authored January 11, 2024
Add support for dynamically computed grid bounds in Pallas kernels.
Authored February 1, 2024
revise logic for tangent types of extended dtypes
Authored December 20, 2023
jnp.unique: make return_inverse shape match NumPy 2.0
Authored January 11, 2024
jnp.linalg: add matrix_norm, matrix_transpose, vector_norm, vector_transpose
Authored December 15, 2023
[Pallas] Add interpret mode support for dynamic grid
Authored February 3, 2024
tweaks to enable adding custom tangent dtypes
Authored December 22, 2023
[ROCm]: Lower sparse(some) ops correctly for ROCm
Authored December 13, 2023
[export] Fix the serialization of effects
Authored December 13, 2023
[shard-map] improve error message when a custom_vjp bwd has extra psum
Authored January 2, 2024
Delete sharding spec to HloSharding conversion since it's not used anymore.
Authored January 2, 2024
[Pallas] Use AbstractMemoryRefs for all Pallas tracing.
Authored January 19, 2024
Always flatten args and kwargs together i.e. `tree_flatten((args, kwargs))` so that we have a uniform in_tree structure everywhere.
Authored December 20, 2023
Add jnp.isdtype function, following np.isdtype in NumPy 2.0
Authored January 17, 2024
[key reuse] handle polymorphic shapes in slice
Authored January 29, 2024
[shape_poly] Improve the lexicographic ordering of symbolic expressions
Authored January 9, 2024
Lower a subset of math primitives directly to Triton IR
Authored January 24, 2024
array api: simplify some wrappers
Authored January 10, 2024
[export] Fix handling of float0 when exporting
Authored December 16, 2023
Implement a stable serialization API for Mosaic
Authored January 2, 2024
[shape_poly] Cache the hash values for symbolic expressions
Authored January 5, 2024
[Mosaic] Simplify the handling of dynamic indices in vector.load and store
Authored January 11, 2024

Browse Other Releases

Latest Pending Unreleased 😎
jax-v0.4.26 Released April 3, 2024
0 Ξ”
jax-v0.4.26-rc Released April 3, 2024
17,086 Ξ”
jax-v0.4.25 Released February 24, 2024
0 Ξ”
jax-v0.4.25-rc Released February 24, 2024
17,651 Ξ”
jax-v0.4.24 Released February 5, 2024
0 Ξ”
jax-v0.4.24-rc Released February 5, 2024
30,241 Ξ”
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 Ξ”
jax-v0.4.22-rc2 Released December 13, 2023
132 Ξ”

Top Contributors in jax-v0.4.24-rc

mattjj
jakevdp
gnecula
superbobry
8bitmp3
sharadmv
yashk2810
apaszke
tlongeri
a-googler

Directory Browser for jax-v0.4.24-rc

All files are compared to previous version, jax-v0.4.23. 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