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

Jax: jaxlib-v0.4.6 Release

Release date:
March 9, 2023
Previous version:
jaxlib-v0.4.4 (released February 16, 2023)
Magnitude:
17,010 Diff Delta
Contributors:
36 total committers
Data confidence:
Commits:
201 impacting commits

201 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
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
Roll back #14792
Authored March 8, 2023
[shape_poly, call_tf] Some improvements for call_tf in a shape polymorphic program
Authored March 1, 2023
Shorten alias chains for names exported in jax. namespace.
Authored March 1, 2023
[shard-map] add systematic tests for eager, jit, autodiff
Authored February 15, 2023
custom_jvp symbolic zeros support
Authored February 17, 2023
[jax2tf] Add support for cross-platform lowering in native serialization
Authored February 20, 2023
attach debug info to jaxpr, pass to mlir/mhlo
Authored March 2, 2023
[shape_poly, call_tf] Some improvements for call_tf in a shape polymorphic program
Authored March 8, 2023
[sparse] bring sparse.csr API in line with sparse.coo
Authored February 16, 2023
A different way to achieve cross-platform lowering, withouth any
Authored February 26, 2023
add result info to mhlo, fixes #14780
Authored March 7, 2023
roll back to avoid weakref constraints
Authored March 2, 2023
lax_vmap_test: Extend timeout so that the TPU variant can run in ASAN.
Authored March 7, 2023
[jax2tf] Clean up the support for cross-lowering.
Authored February 28, 2023
Add sharding to the signature of shard_args and update
Authored March 2, 2023
Internal change
Authored March 7, 2023
[sparse] modify impl rules to always use lowering path
Authored February 16, 2023
Remove more exported names from jax.interpreters.xla.
Authored February 16, 2023
Split some submodules out of //jax under Bazel.
Authored March 9, 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
Split _src/traceback_util.py into its own Bazel target.
Authored March 9, 2023
[sparse] test coo/csr extra nse
Authored February 17, 2023
[sparse] fix GPU warnings in cusparse test
Authored March 7, 2023
jnp.ndarray.view: implement all dtypes
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
[Rollforward] Convert _arrays to return PyArray instead of PyBuffer.
Authored February 17, 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
[shard_map] make debug_print work with shard_map, eager and jit
Authored March 9, 2023
[sparse] simplify bcoo_dot_general GPU lowering rule
Authored March 2, 2023
Fix nondeterminism issue with ordered effects
Authored February 24, 2023
[sparse] improve worst-case nse in spdot_general
Authored March 8, 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
jnp.ndarray.view: implement all dtypes
Authored February 16, 2023

Browse Other Releases

Latest Pending Unreleased 😎
0.3.24 Released November 4, 2022
23,084 Ξ”
0.3.23 Released October 12, 2022
305 Ξ”
0.3.22 Released October 11, 2022
11,644 Ξ”
0.3.21 Released September 30, 2022
1,561 Ξ”
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 Ξ”

Top Contributors in jaxlib-v0.4.6

jake-vanderplas-1c1d
mattjj
hawkinsp
gnecula
sharadmv
yashk2810
pschuh
pizzud
a-googler
skye

Directory Browser for jaxlib-v0.4.6

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