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
1
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:
21,059 Diff Delta
Contributors:
36 total committers
Data confidence:
Commits:
202 impacting commits

202 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
Roll back #14792
Authored March 8, 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
attach debug info to jaxpr, pass to mlir/mhlo
Authored March 2, 2023
[jax2tf] Clean up the support for cross-lowering
Authored February 28, 2023
roll back to avoid weakref constraints
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
lax_vmap_test: Extend timeout so that the TPU variant can run in ASAN
Authored March 7, 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
Split _src/traceback_util.py into its own Bazel target
Authored March 9, 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
Remove more exported names from jax.interpreters.xla
Authored February 16, 2023
[shard-map] add annotations and notes to shard_map_test.py
Authored February 17, 2023
[sparse] test coo/csr extra nse
Authored February 17, 2023
[sparse] fix GPU warnings in cusparse test
Authored March 7, 2023
Rename `jax.sharding.OpShardingSharding` to `jax.sharding.GSPMDSharding`
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] make debug_print work with shard_map, eager and jit
Authored March 9, 2023
jnp.ndarray.view: implement all dtypes
Authored February 17, 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
[sparse] remove handling of padded indices from COO/CSR
Authored February 22, 2023
jnp.ndarray.view: implement all dtypes
Authored February 16, 2023

Browse Other Releases

Latest Pending Unreleased 😎
jaxlib-v0.4.11 Released June 3, 2023
7,496 Ξ”
jaxlib-v0.4.10 Released June 3, 2023
1,240 Ξ”
jaxlib-v0.4.9 Released June 3, 2023
26,712 Ξ”
jaxlib-v0.4.7 Released March 27, 2023
28,303 Ξ”
jaxlib-v0.4.6 Released March 9, 2023
21,059 Ξ”
jaxlib-v0.4.4 Released February 16, 2023
6,670 Ξ”
jaxlib-v0.4.3 Released February 8, 2023
14,421 Ξ”
jaxlib-v0.4.2 Released January 25, 2023
31,232 Ξ”
jaxlib-v0.4.1 Released December 13, 2022
592 Ξ”
jaxlib-v0.4.0 Released December 12, 2022
20,648 Ξ”

Top Contributors in jaxlib-v0.4.6

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

Directory Browser for jaxlib-v0.4.6

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