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
0
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:
24,737 Diff Delta
Contributors:
44 total committers
Data confidence:
Commits:
288 impacting commits

288 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
[jax2tf] Adds tests for polymorphic input for Flax Module Harnesses.
Authored January 10, 2023
[jax2tf] Refactoring of shape_poly_test.
Authored December 17, 2022
Checkify: switch to initial-style.
Authored December 6, 2022
[shape_poly] Generalize binary operations with symbolic dimensions.
Authored January 18, 2023
[sparse] refactor/clean up test suite
Authored December 16, 2022
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
Expose fp8 in jax dtypes and mlir builder.
Authored January 14, 2023
[jax2tf] Improve handling of dimension polynomials used with JAX arrays.
Authored December 28, 2022
Make `jit` a thin wrapper around `pjit` which ignores the mesh context manager (just like how it is today)
Authored January 13, 2023
Support `static_argnames` in pjit as a first step to merge `jit` and `pjit`.
Authored December 13, 2022
[jax2tf] Ensure that dim_as_value returns int64 in x64 mode and int32 otherwise
Authored December 20, 2022
Style changes and typo corrections
Authored December 16, 2022
[sparse] Add bcsr dot_general
Authored December 19, 2022
Make the device assignment of outfeed configurable
Authored December 20, 2022
[sparse] add higher-level version of bcoo_extract & improve tests
Authored January 5, 2023
Fix scan and name_stack, rewrite cond to use jaxpr_to_checkify_jaxpr.
Authored December 20, 2022
[jax2tf] Improves support for examples testing and adds three examples.
Authored January 9, 2023
[sparse] bcoo_extract: add assume_unique keyword
Authored January 4, 2023
Port type inference for 6 ops from StableHLO to MHLO
Authored December 14, 2022
[sparse] handle general batch dimensions in bcoo transpose
Authored December 23, 2022
[jax2tf] Improves jax2tf (enable_xla=False) model testing logic.
Authored January 5, 2023
[jax2tf] Improve handling of lax.pad and jnp.pad with polymorphic padding config
Authored December 29, 2022
Sharp Bits: add section on Dynamic shapes
Authored January 19, 2023
[sparse] Use SparseInfo for both bcoo and bcsr format.
Authored December 16, 2022
[host_callback] Add device_index to hcb.call and add tests
Authored January 18, 2023
Add checkify.debug_check which is a noop outside of checkify.
Authored December 12, 2022
Migrate pytestmark usage to new `@jtu.pytest_mark_if_available` decorator.
Authored January 12, 2023
[jax2tf] Fixed the conversion of a function that contains an inner pjit
Authored January 8, 2023
[sparse] add CheckGradsSparse test utility
Authored December 15, 2022
[jax2tf] Adjust jax2tf for the pjit==jit API migration.
Authored January 17, 2023
Lift global_axis calculation from lowering in pxla.py to api.py.
Authored January 5, 2023
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. Also remove usage of `xc._version` and replace it with `xla_extension_version`.
Authored December 19, 2022
Error on numpy masked array inputs.
Authored December 27, 2022
Sharp bits: raise exceptions directly
Authored January 17, 2023
[sparse] generalize batch rule for bcoo_spdot_general
Authored January 6, 2023
Introduce `is_equivalent_to` method on Sharding to check if 2 shardings mean the same thing.
Authored January 23, 2023

Browse Other Releases

Latest Pending Unreleased 😎
0.3.20 Released March 9, 2023
101,633 Ξ”
jaxlib-v0.4.6 Released March 9, 2023
17,010 Ξ”
jaxlib-v0.4.4 Released February 16, 2023
6,724 Ξ”
jaxlib-v0.4.3 Released February 8, 2023
7,645 Ξ”
jaxlib-v0.4.2 Released January 25, 2023
24,737 Ξ”
jaxlib-v0.4.1 Released December 13, 2022
596 Ξ”
jaxlib-v0.4.0 Released December 12, 2022
16,803 Ξ”
jaxlib-v0.3.25 Released November 15, 2022
6,881 Ξ”
jaxlib-v0.3.24 Released November 4, 2022
23,948 Ξ”
jaxlib-v0.3.22 Released October 11, 2022
13,198 Ξ”

Top Contributors in jaxlib-v0.4.2

jake-vanderplas-1c1d
yashk2810
gnecula
seanprime7
burmako
marcvanzee
LenaMartens
zhangqiaorjc
tlu7
sharadmv

Directory Browser for jaxlib-v0.4.2

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