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

Jax: jaxlib-v0.4.2 Release

Release date:
January 25, 2023
Previous version:
jaxlib-v0.4.1 (released December 13, 2022)
Magnitude:
25,762 Diff Delta
Contributors:
44 total committers
Data confidence:
Commits:
297 impacting commits

297 Commits in this Release

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

Convert the notebook to a plain MD file
Authored December 15, 2022
(NFC) Prepare for migration from producing MHLO to producing StableHLO
Authored December 16, 2022
Delete the pjit 101 tutorial which is subsumed by https://jax.readthedocs.io/en/latest/notebooks/Distributed_arrays_and_automatic_parallelization.html
Authored January 21, 2023
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
migrate internal dependencies from `jax.core` to `jax._src.core`
Authored December 17, 2022
Instructions to create custom operation for GPUs
Authored December 13, 2022
Checkify: switch to initial-style
Authored December 6, 2022
[jax2tf] Refactoring of shape_poly_test
Authored December 17, 2022
[sparse] add nfold_vmap utility
Authored January 6, 2023
[shape_poly] Generalize binary operations with symbolic dimensions
Authored January 18, 2023
Delete `jax.experimental.callback`
Authored January 13, 2023
Multinodes CICD on GPUs using on-demand cluster and e2e tests using T5X
Authored September 26, 2022
[sparse] refactor/clean up test suite
Authored December 16, 2022
Add IO callback
Authored November 10, 2022
Remove experimental endpoints and update to point to 0.4.1
Authored December 13, 2022
Add `device` and `backend` API to `pjit` but resolve them away in `infer_params`
Authored December 14, 2022
[jax2tf] Improve handling of dimension polynomials used with JAX arrays
Authored December 28, 2022
[sparse] improve implementation of sparse.grad & sparse.value_and_grad
Authored December 28, 2022
Expose fp8 in jax dtypes and mlir builder
Authored January 14, 2023
[jax2tf] Ensure that dim_as_value returns int64 in x64 mode and int32 otherwise
Authored December 20, 2022
Support `static_argnames` in pjit as a first step to merge `jit` and `pjit`
Authored December 13, 2022
Make `jit` a thin wrapper around `pjit` which ignores the mesh context manager (just like how it is today)
Authored January 13, 2023
[sparse] Add bcsr dot_general
Authored December 19, 2022
[sparse] refactor tests to improve runtime
Authored January 20, 2023
[sparse] add higher-level version of bcoo_extract & improve tests
Authored January 5, 2023
[sparse] bcoo_extract: add assume_unique keyword
Authored January 4, 2023
[jax2tf] Improve handling of lax.pad and jnp.pad with polymorphic padding config
Authored December 29, 2022
[sparse] handle general batch dimensions in bcoo transpose
Authored December 23, 2022
[sparse] Use SparseInfo for both bcoo and bcsr format
Authored December 16, 2022
Lift global_axis calculation from lowering in pxla.py to api.py
Authored January 5, 2023
Add checkify.debug_check which is a noop outside of checkify
Authored December 12, 2022
Make the device assignment of outfeed configurable
Authored December 20, 2022
Remove code that existed to support the now-gone classic HLO lowering path
Authored December 20, 2022
Remove more checks now that the minimum jaxlib version corresponds to xla_extension_version == 109
Authored December 19, 2022
[jax2tf] Improves support for examples testing and adds three examples
Authored January 9, 2023
[jax2tf] Improves jax2tf (enable_xla=False) model testing logic
Authored January 5, 2023
Sharp Bits: add section on Dynamic shapes
Authored January 19, 2023
Error on numpy masked array inputs
Authored December 27, 2022
Simplify Array's shard_arg_handler by merging pmap and pjit/xmap paths
Authored December 27, 2022

Browse Other Releases

Latest Pending Unreleased 😎
jaxlib-v0.4.7 Released March 27, 2023
24,447 Ξ”
jaxlib-v0.4.6 Released March 9, 2023
18,194 Ξ”
jaxlib-v0.4.4 Released February 16, 2023
5,774 Ξ”
jaxlib-v0.4.3 Released February 8, 2023
12,125 Ξ”
jaxlib-v0.4.2 Released January 25, 2023
25,762 Ξ”
jaxlib-v0.4.1 Released December 13, 2022
501 Ξ”
jaxlib-v0.4.0 Released December 12, 2022
16,767 Ξ”
jaxlib-v0.3.25 Released November 15, 2022
6,770 Ξ”
jaxlib-v0.3.24 Released November 4, 2022
20,656 Ξ”
jaxlib-v0.3.22 Released October 11, 2022
16 Ξ”

Top Contributors in jaxlib-v0.4.2

yashk2810
jakevdp
gnecula
seanprime7
burmako
hawkinsp
marcvanzee
LenaMartens
sharadmv
froystig

Directory Browser for jaxlib-v0.4.2

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

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