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

Jax: jax-v0.4.0-rc1 Release

Release date:
December 12, 2022
Previous version:
jax-v0.4.0-rc (released December 8, 2022)
Magnitude:
2,877 Diff Delta
Contributors:
11 total committers
Data confidence:
Commits:
21 impacting commits

21 Commits in this Release

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

[jax2tf] An alternative support for shape polymorphism for native serialization.
Authored November 28, 2022
DOC: add example of pure_callback with custom_jvp
Authored December 9, 2022
Improves handling of opaque types for dynamic shapes
Authored December 4, 2022
in jax.Array notebook, polish beginning and tweak title and some wording
Authored December 11, 2022
Make JAX tests pass under NumPy 1.24.0rc2.
Authored December 8, 2022
[sparse] fix logic bug in GPU testcase
Authored December 12, 2022
Checkify: Add documentation for adding run-time values to error message.
Authored December 9, 2022
Add `jax.scipy.linalg.toeplitz`.
Authored November 15, 2022
[sparse] Add rand_bcsr to generate a random BCSR array.
Authored December 9, 2022
Move `with_sharding_constraint` out of experimental into `jax.lax` namespace.
Authored December 12, 2022
[sparse] Move _dot_general_validated_shape to sparse util.
Authored December 9, 2022
Add device_ids and axis_names to the Mesh repr
Authored December 8, 2022
[sparse] Fix a bug in BCSR tree_flatten.
Authored December 9, 2022
flake8 now rejects inline comments.
Authored December 8, 2022
[jax2tf] Fix for jnp.roll with shape polymorphism
Authored December 7, 2022
Add workaround for imprecise shape inference for DynamicGatherOp
Authored December 12, 2022
[TPU CI] Run build matrix on v3-8 as well as v4-8
Authored December 9, 2022
Fail lower_jaxpr_to_module if the module fails verification
Authored December 8, 2022
Expose channel_id in AllToAllOp in both XLA builder and MHLO.
Authored December 10, 2022
Small fix to scan type-check error message.
Authored December 8, 2022
Set bcoo_cusparse_lowering to False by default
Authored December 9, 2022

Browse Other Releases

Latest Pending Unreleased 😎
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 Ξ”
jax-v0.3.25 Released November 15, 2022
0 Ξ”
jax-v0.3.25-rc3 Released November 15, 2022
28 Ξ”
jax-v0.3.25-rc2 Released November 15, 2022
3 Ξ”
jax-v0.3.25-rc1 Released November 15, 2022
3 Ξ”

Top Contributors in jax-v0.4.0-rc1

gnecula
jakevdp
hawkinsp
mattjj
tlu7
LenaMartens
yashk2810
yotarok
skye
burmako

Directory Browser for jax-v0.4.0-rc1

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