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

Jax: jax-v0.4.2-rc Release

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

290 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
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
[jax2tf] Refactoring of shape_poly_test.
Authored December 17, 2022
migrate internal dependencies from `jax.core` to `jax._src.core`
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
[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
[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 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
Remove code that existed to support the now-gone classic HLO lowering path.
Authored December 20, 2022
Error on numpy masked array inputs.
Authored December 27, 2022
Sharp bits: raise exceptions directly
Authored January 17, 2023
Port type inference for 6 ops from StableHLO to MHLO
Authored December 14, 2022
[sparse] generalize batch rule for bcoo_spdot_general
Authored January 6, 2023

Browse Other Releases

Latest Pending Unreleased 😎
jax-v0.4.4-rc Released February 16, 2023
6,660 Ξ”
jax-v0.4.3 Released February 8, 2023
0 Ξ”
jax-v0.4.3-rc Released February 8, 2023
8,565 Ξ”
jax-v0.4.2 Released January 25, 2023
0 Ξ”
jax-v0.4.2-rc Released January 25, 2023
26,261 Ξ”
jax-v0.4.1 Released December 13, 2022
0 Ξ”
jax-v0.4.1-rc Released December 13, 2022
592 Ξ”
jax-v0.4.0 Released December 12, 2022
0 Ξ”
jax-v0.4.0-rc1 Released December 12, 2022
2,877 Ξ”
jax-v0.4.0-rc Released December 8, 2022
14,191 Ξ”

Top Contributors in jax-v0.4.2-rc

jakevdp
yashk2810
gnecula
seanprime7
burmako
marcvanzee
hawkinsp
LenaMartens
froystig
zhangqiaorjc

Directory Browser for jax-v0.4.2-rc

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