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

Jax: jaxlib-v0.1.33 Release

Release date:
April 3, 2024
Previous version:
jax-v0.4.26 (released April 3, 2024)
Magnitude:
458,289 Diff Delta
Contributors:
358 total committers
Data confidence:
Commits:
6,889 impacting commits

6889 Commits in this Release

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

Migrate remaining tests from jtu.cases_from_list to jtu.sample_product.
Authored October 12, 2022
Upgrade JAX Advanced Autodiff 201
Authored December 15, 2023
[shard-map] add user tutorial
Authored January 8, 2024
Merge CUDA and ROCM kernel code in jaxlib.
Authored October 25, 2022
Apply pyupgrade --py39-plus.
Authored July 21, 2023
Update JAX to use new math libraries in ROCm-5.0.
Authored February 15, 2022
prototyping dynamic shapes
Authored March 31, 2022
Rewrite Checkify to support tracking different error types.
Authored October 18, 2022
(NFC) Prepare for migration from producing MHLO to producing StableHLO
Authored December 16, 2022
Update flags to use the ABSL typed flag API.
Authored July 27, 2023
roll-forward #11952
Authored August 20, 2022
introduce key-element-type arrays and overhaul the Python PRNG key array type
Authored August 10, 2022
Refactor effects system to use effect types, not objects
Authored February 2, 2023
[shape_poly] Copy many of the jax2tf/shape_poly_test to live outside of jax2tf.
Authored November 19, 2023
Add initial implementation of a `run_state` primitive
Authored February 17, 2023
Add an internal jtu.sample_product test decorator.
Authored October 3, 2022
Added ir.Value-based versions of some triton.compat APIs
Authored February 13, 2024
Add debugging tools under `jax.debug` and documentation
Authored July 26, 2022
remove experimental/djax
Authored April 8, 2022
[jax2tf] Rewrites `converters_eval` framework.
Authored September 6, 2022
Update minimum jaxlib version to 0.3.14.
Authored July 8, 2022
[Pallas] Refactor indexing primitives to use NDIndexer abstraction
Authored January 2, 2024
Bump the minimum jaxlib version to 0.4.1.
Authored December 19, 2022
[jax2tf] Adds tests for polymorphic input for Flax Module Harnesses.
Authored January 10, 2023
[shape_poly] Add limited support for equality explicit constraints.
Authored January 27, 2024
djax: let make_jaxpr build dyn shape jaxprs
Authored January 21, 2022
Add Mosaic to Jaxlib and expose bindings in `jax.experimental.mosaic`
Authored July 21, 2023
[shape_poly] Cleaning up naming of terms and factors.
Authored February 20, 2024
Remove GDA tests from JAX since GDA is deprecated. There are jax.Array tests for all the corresponding GDA tests
Authored March 15, 2023
[shape_poly] Add a decision procedure for inequalities.
Authored January 20, 2024
[shape_poly] Improve compile-time shape checking.
Authored May 27, 2023
Pallas now uses MLIR Python builders to lower to Triton IR
Authored January 11, 2024
[jax2tf] An alternative support for shape polymorphism for native serialization.
Authored November 28, 2022
Add CUDNN custom call for LSTM. Exposed as jax.experimental.rnn module.
Authored November 23, 2022
Upgrade JAX pytrees doc
Authored December 7, 2023
feat: refactor code using pyupgrade
Authored May 12, 2022
Split jax.interpreters.xla up into three pieces:
Authored November 22, 2021
Add initial LOBPCG top-k eigenvalue solver (#3112)
Authored June 2, 2022
Merge remaining CUDA and ROCM Python code.
Authored May 6, 2022
Bump minimum jaxlib_version to 0.4.2 i.e xla_extension_version == 119 and mlir_api_version == 43
Authored February 6, 2023

Browse Other Releases

Latest Pending Unreleased 😎
jaxlib-v0.1.39 Released February 14, 2020
0 Ξ”
jaxlib-v0.1.38 Released January 29, 2020
0 Ξ”
jaxlib-v0.1.37 Released December 6, 2019
0 Ξ”
jaxlib-v0.1.36 Released November 22, 2019
0 Ξ”
jaxlib-v0.1.33 Released April 3, 2024
458,289 Ξ”
jax-v0.4.28 Released May 9, 2024
0 Ξ”
jax-v0.4.28-rc Released May 9, 2024
0 Ξ”
jax-v0.4.27 Released May 7, 2024
0 Ξ”
jax-v0.4.27-rc Released May 7, 2024
18,400 Ξ”
jax-v0.4.26 Released April 3, 2024
0 Ξ”

Top Contributors in jaxlib-v0.1.33

hawkinsp
jakevdp
yashk2810
gnecula
mattjj
sharadmv
froystig
a-googler
superbobry
LenaMartens

Directory Browser for jaxlib-v0.1.33

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