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

Jax: jaxlib-v0.4.24 Release

Release date:
February 5, 2024
Previous version:
jaxlib-v0.4.23 (released December 14, 2023)
Magnitude:
30,896 Diff Delta
Contributors:
50 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
Rename _wraps to implements
Authored January 24, 2024
Upgrade JAX primitives doc
Authored January 23, 2024
Pallas now uses MLIR Python builders to lower to Triton IR
Authored January 11, 2024
Add nested pipeline/pallas_call support for TPU meta-programming of collectives + compute
Authored January 19, 2024
[shape_poly] Add a decision procedure for inequalities
Authored January 20, 2024
[Pallas/Mosaic] Add support for nested `ref.at`
Authored January 3, 2024
[shape_poly] Add more tests for reasoning about inequalities
Authored January 24, 2024
Upgrade JAX internals 301 jaxpr language tutorial
Authored December 30, 2023
[Pallas/TPU] Add pallas call tests
Authored January 19, 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
Remove a number of deprecated APIs
Authored December 18, 2023
[shape_poly] Fixes for the lexicographic ordering of monomials
Authored January 5, 2024
Inlined all builder.create_* calls
Authored February 5, 2024
jax.random: deprecate passing of batched keys to APIs
Authored January 17, 2024
Expose `shard_alike` via jax.experimental. 0
Authored December 20, 2023
Remove indices and devices from shard_arg_handlers and shard_args
Authored January 5, 2024
Ported a subset of binary operations to lower directly to Triton IR
Authored January 26, 2024
Generate Python bindings for the Triton MLIR dialect
Authored January 2, 2024
add jax.explain_cache_misses tracing cache miss explanations
Authored June 9, 2023
Cache ClosedJaxpr creation to minimize cache_misses. 0
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
revise logic for tangent types of extended dtypes
Authored December 20, 2023
jnp.linalg: add matrix_norm, matrix_transpose, vector_norm, vector_transpose
Authored December 15, 2023
[ROCm]: Lower sparse(some) ops correctly for ROCm
Authored December 13, 2023
[Pallas] Use AbstractMemoryRefs for all Pallas tracing
Authored January 19, 2024
Lower a subset of math primitives directly to Triton IR
Authored January 24, 2024
Always flatten args and kwargs together i.e. 0
Authored December 20, 2023
array api: simplify some wrappers
Authored January 10, 2024
Add jnp.isdtype function, following np.isdtype in NumPy 2.0
Authored January 17, 2024
Delete sharding spec to HloSharding conversion since it's not used anymore
Authored January 2, 2024
[Pallas] Add interpret mode support for dynamic grid
Authored February 3, 2024
[export] Fix the serialization of effects
Authored December 13, 2023
jnp.unique: make return_inverse shape match NumPy 2.0
Authored January 11, 2024
Implement a stable serialization API for Mosaic
Authored January 2, 2024

Browse Other Releases

Latest Pending Unreleased 😎
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 Ξ”
jaxlib-v0.4.20 Released November 2, 2023
4,475 Ξ”

Top Contributors in jaxlib-v0.4.24

jakevdp
8bitmp3
superbobry
mattjj
gnecula
sharadmv
yashk2810
epiqueras
apaszke
tlongeri

Directory Browser for jaxlib-v0.4.24

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

Release Notes Published

jaxlib release v0.4.24

GitClear uses cookies to ensure you get the best experience on our website. Learn more about our cookie policy