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

Jax: jax-v0.4.6-rc Release

Release date:
March 9, 2023
Previous version:
jax-v0.4.5 (released March 2, 2023)
Magnitude:
5,714 Diff Delta
Contributors:
20 total committers
Data confidence:
Commits:
72 impacting commits

72 Commits in this Release

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

Roll back #14792
Authored March 8, 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
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 some submodules out of //jax under Bazel.
Authored March 9, 2023
Split _src/traceback_util.py into its own Bazel target.
Authored March 9, 2023
[sparse] fix GPU warnings in cusparse test
Authored March 7, 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
[sparse] improve worst-case nse in spdot_general
Authored March 8, 2023
[sparse] more robust correction for out-of-bound indices in BCOO
Authored March 2, 2023
[shape_poly] Add support for jnp.cum{sum,prod,max,min} with shape polymorphism
Authored February 27, 2023
Change definition of util.wraps so pytype can understand it.
Authored March 3, 2023
Internal change
Authored March 4, 2023
update relu6 grad at 0 and 6 to match pytorch convention
Authored March 8, 2023
[shape_poly] Add support for shape polymorphism for jnp.{argsort,bincount,insert,nonzero}
Authored March 2, 2023
[shape_poly, call_tf] Some improvements for call_tf in a shape polymorphic program
Authored March 1, 2023
Fix a TODO updating users of lax._check_user_dtype_supported to dtypes.check_user_dtype_supported.
Authored March 6, 2023
Add "compare_all" method to searchsorted
Authored March 7, 2023
deprecation_module: Move to new internal_test_util directory.
Authored February 17, 2023
DOC: clarify variable names in scan doc
Authored March 2, 2023
Split _src/pretty_printer.py into its own Bazel target.
Authored March 9, 2023
Set ArrayImpl.__name__ to ArrayImpl
Authored March 8, 2023
add key path change to changelog
Authored March 4, 2023
Retrace jaxpr again if the same function with the same avals is run under different meshes. Add pxla.thread_resources to the global trace_context which takes care of the caching of global variables.
Authored March 6, 2023
Reenable pytype for numpy ufuncs.
Authored March 3, 2023
Add argnames and resultnames to pmap.
Authored March 8, 2023
[sparse] add low-level primitives wrapping cuda csr spmv and spmm.
Authored March 6, 2023
Convert shard_args to return arrays when jax.config.jax_array is True.
Authored March 9, 2023
[sparse] add support for integer_pow
Authored March 8, 2023
[sparse] more careful handling of index_dtype in rand_bcoo
Authored March 7, 2023
Fix the case where debug_info was not attached when a xmap was present in the computation.
Authored March 3, 2023
Convert testShardArgs to handle pxla.Chunked sharding properly.
Authored March 7, 2023
Remove the disassemble into single devices arrays in ExecuteReplicated.__call__ since it is not needed anymore since shard_args return Array only with `xla_extension_version >= 136`.
Authored March 9, 2023

Browse Other Releases

Latest Pending Unreleased 😎
jax-v0.4.8 Released March 29, 2023
2,231 Ξ”
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 Ξ”

Top Contributors in jax-v0.4.6-rc

mattjj
gnecula
jakevdp
pschuh
hawkinsp
pizzud
yashk2810
a-googler
shoyer
IvyZX

Directory Browser for jax-v0.4.6-rc

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