Home » Oreps » Jax » All Releases » jax-v0.4.5 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.5
Basic Details Issues Addressed Top Contributors Directory Browser Release Notes
Jax jax-v0.4.5

Jax: jax-v0.4.5 Release

Release date:
March 2, 2023
Previous version:
jax-v0.4.4 (released February 16, 2023)
Magnitude:
11,708 Diff Delta
Contributors:
27 total committers
Data confidence:
Commits:
129 impacting commits

129 Commits in this Release

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

Refactor effects system to use effect types, not objects
Authored February 2, 2023
Add `JaxprInputEffect` and refactor `StateEffect`s to use it
Authored February 17, 2023
Replace jax._src.util.prod with math.prod.
Authored February 28, 2023
Refactor `Ref` abstract type to contain other `AbstractValue`s
Authored February 17, 2023
[shard-map] add systematic tests for eager, jit, autodiff
Authored February 15, 2023
Shorten alias chains for names exported in jax. namespace.
Authored March 1, 2023
[jax2tf] Add support for cross-platform lowering in native serialization
Authored February 20, 2023
custom_jvp symbolic zeros support
Authored February 17, 2023
A different way to achieve cross-platform lowering, withouth any
Authored February 26, 2023
[sparse] bring sparse.csr API in line with sparse.coo
Authored February 16, 2023
Add support for XLAGather with 2-D Batch Dimensions for `enable_xla=False`
Authored February 23, 2023
Replace usage of {in|out}_axis_resources with {in|out}_shardings
Authored February 28, 2023
Remove more exported names from jax.interpreters.xla.
Authored February 16, 2023
[sparse] test coo/csr extra nse
Authored February 17, 2023
Rename `jax.sharding.OpShardingSharding` to `jax.sharding.GSPMDSharding`. `jax.sharding.OpShardingSharding` will be removed in 3 months from Feb 17, 2023.
Authored February 18, 2023
lax_test: Create a separate module for lax-specific test utils in a new package.
Authored February 16, 2023
Delete `PyShardedBuffer` since it was only used for GDAs and GDA is deprecated.
Authored February 21, 2023
[shard-map] add annotations and notes to shard_map_test.py
Authored February 17, 2023
jnp.ndarray.view: implement all dtypes
Authored February 17, 2023
Fix nondeterminism issue with ordered effects
Authored February 24, 2023
[sparse] bcoo_dot_general_sampled: another special case
Authored February 24, 2023
lax.bitcast_convert_type: support casting between types of different width
Authored February 16, 2023
[sparse] remove handling of padded indices from COO/CSR
Authored February 22, 2023
jnp.ndarray.view: implement all dtypes
Authored February 16, 2023
Add back opt-barrier fallback, since the fallback sometimes prevents OOMs.
Authored February 7, 2023
add random.chisquare and random.f
Authored February 24, 2023
Move global_device_array into its own BUILD target.
Authored February 16, 2023
Remove use_stablehlo as minimum mlir_api_version >= 43
Authored February 24, 2023
[jax2tf] Enable strict platform checking for native serialized modules.
Authored February 21, 2023
Roll-back https://github.com/google/jax/pull/14526 because it breaks `view()` on scalar inputs
Authored February 17, 2023
[mhlo] Use XLA pretty-printed format for shardingattr
Authored March 1, 2023
Pass the `jaxpr` from `pjit` since there is no need to trace it again in lower_sharding_computation. It also helps in preserving debug_info that already exists on the jaxpr to surface it in MHLO eventually.
Authored March 1, 2023
[jax2tf] Customize limitations for native lowering mode
Authored February 21, 2023
[custom_vjp] bwd function should not be WrappedFun, may run multiple times
Authored February 28, 2023
[sparse] bcoo_dot_general_sampled: faster special case
Authored February 22, 2023
DOC: improve usage recommendation in jax.typing
Authored February 21, 2023
fixes from reviewers
Authored February 22, 2023
[sparse] add low-level primitives wrapping cuda SpMV & SpMM
Authored March 1, 2023
Add PyArrayResultHandler which behaves like
Authored February 28, 2023
Add "Open in Kaggle" buttons to Jupyter notebooks.
Authored March 1, 2023

Browse Other Releases

Latest Pending Unreleased 😎
jax-v0.4.7 Released March 27, 2023
0 Ξ”
jax-v0.4.7-rc1 Released March 27, 2023
0 Ξ”
jax-v0.4.7-rc Released March 27, 2023
19,399 Ξ”
jax-v0.4.6 Released March 9, 2023
0 Ξ”
jax-v0.4.6-rc Released March 9, 2023
5,714 Ξ”
jax-v0.4.5 Released March 2, 2023
11,708 Ξ”
jax-v0.4.4 Released February 16, 2023
0 Ξ”
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 Ξ”

Top Contributors in jax-v0.4.5

sharadmv
jakevdp
hawkinsp
yashk2810
mattjj
gnecula
a-googler
pschuh
pizzud
atondwal

Directory Browser for jax-v0.4.5

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

Release Notes Published

jax version 0.4.5

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