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
Pull Requests
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:
18,194 Diff Delta
Contributors:
36 total committers
Data confidence:
Commits:
205 impacting commits

205 Commits in this Release

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

internal: avoid unused imports in lax_numpy
Authored March 8, 2023
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
add remat tutorial docs
Authored December 21, 2022
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
Bump minimum jaxlib version to 0.4.4 which means xla_extension_version >= 127
Authored February 24, 2023
Shorten alias chains for names exported in jax. namespace
Authored March 1, 2023
Roll back #14792
Authored March 8, 2023
attach debug info to jaxpr, pass to mlir/mhlo
Authored March 2, 2023
[jax2tf] Add support for cross-platform lowering in native serialization
Authored February 20, 2023
roll back to avoid weakref constraints
Authored March 2, 2023
[jax2tf] Clean up the support for cross-lowering
Authored February 28, 2023
[shard-map] add systematic tests for eager, jit, autodiff
Authored February 15, 2023
custom_jvp symbolic zeros support
Authored February 17, 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
Add sharding to the signature of shard_args and update
Authored March 2, 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
lax_vmap_test: Extend timeout so that the TPU variant can run in ASAN
Authored March 7, 2023
Split some submodules out of //jax under Bazel
Authored March 9, 2023
Internal change
Authored March 7, 2023
Remove more exported names from jax.interpreters.xla
Authored February 16, 2023
Delete `PyShardedBuffer` since it was only used for GDAs and GDA is deprecated
Authored February 21, 2023
Rename `jax.sharding.OpShardingSharding` to `jax.sharding.GSPMDSharding`
Authored February 18, 2023
[sparse] test coo/csr extra nse
Authored February 17, 2023
[shard_map] make debug_print work with shard_map, eager and jit
Authored March 9, 2023
Replace usage of {in|out}_axis_resources with {in|out}_shardings
Authored February 28, 2023
[sparse] simplify bcoo_dot_general GPU lowering rule
Authored March 2, 2023
jnp.ndarray.view: implement all dtypes
Authored February 17, 2023
Add support for XLAGather with 2-D Batch Dimensions for `enable_xla=False`
Authored February 23, 2023
[shard-map] add annotations and notes to shard_map_test.py
Authored February 17, 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
[sparse] remove handling of padded indices from COO/CSR
Authored February 22, 2023
Remove use_stablehlo as minimum mlir_api_version >= 43
Authored February 24, 2023

Browse Other Releases

Latest Pending Unreleased 😎
jaxlib-v0.4.11 Released June 3, 2023
6,363 Ξ”
jaxlib-v0.4.10 Released June 3, 2023
1,057 Ξ”
jaxlib-v0.4.9 Released June 3, 2023
23,085 Ξ”
jaxlib-v0.4.7 Released March 27, 2023
24,447 Ξ”
jaxlib-v0.4.6 Released March 9, 2023
18,194 Ξ”
jaxlib-v0.4.4 Released February 16, 2023
5,774 Ξ”
jaxlib-v0.4.3 Released February 8, 2023
12,125 Ξ”
jaxlib-v0.4.2 Released January 25, 2023
25,762 Ξ”
jaxlib-v0.4.1 Released December 13, 2022
501 Ξ”
jaxlib-v0.4.0 Released December 12, 2022
16,767 Ξ”

Top Contributors in jaxlib-v0.4.6

jakevdp
mattjj
sharadmv
hawkinsp
yashk2810
gnecula
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.

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