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

Jax: jax-v0.4.20-rc Release

Release date:
November 2, 2023
Previous version:
jax-v0.4.19 (released October 18, 2023)
Magnitude:
5,123 Diff Delta
Contributors:
26 total committers
Data confidence:
Commits:
103 impacting commits

103 Commits in this Release

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

output res forwarding optimization for shard_map and jit
Authored October 12, 2023
[Mosaic] Add support for (almost) arbitrary reductions
Authored October 26, 2023
draft docs on gpu performance tuning
Authored October 17, 2023
Deprecated `define_*` and `DEFINE_*` methods on `jax.config`
Authored October 25, 2023
Cleanup the handling of single- and multi-platform lowering in ModuleContext
Authored October 25, 2023
Documentation: add stub files for new tutorial structure
Authored October 30, 2023
[typing] regularize types of jax.random API
Authored October 18, 2023
[Pallas TPU] Add support for hoisted scratch spaces
Authored October 25, 2023
[jex] replace `extend.random.PRNGImpl` with `extend.random.define_prng_impl`
Authored November 1, 2023
[export] Add jax.global_constant MLIR attributes for dimension variable arguments
Authored October 20, 2023
[call_tf] Fix call_tf lowering for multi-platform lowering
Authored October 13, 2023
[export] Improve the calling of multi-platform exported module
Authored October 18, 2023
[random] make PRNG impl attributes private
Authored October 17, 2023
MAINT Do not use `str()` and `repr()` in f-string replacement fields
Authored October 23, 2023
factor out subs_list and subs_list2
Authored October 19, 2023
Properly pack and unpack int4 arrays on CPU in PJRT.
Authored November 2, 2023
[Try again] For nested pjit's cache the generation of StableHLO if it satifies the key. This should help in improving the lowering time.
Authored October 30, 2023
MAINT Do not import the config object in JAX internals
Authored October 18, 2023
Add jax.numpy.fill_diagonal.
Authored October 20, 2023
[random] deprecate jax.random.default_prng_impl()
Authored October 19, 2023
Reverts 9b1a656c1ef1f93f9e93eccb662de9bebe66b51a
Authored October 24, 2023
[pallas:gpu] Lower more complex primitives using JAX functions in terms of more basic primitives.
Authored October 23, 2023
Add nightly install instructions to docs
Authored October 26, 2023
[IFRT] Fix incorrect type numbers for e4m3 and e5m2 types.
Authored October 24, 2023
Remove bitwise conversions from _canonicalize_float_for_sort
Authored October 25, 2023
Cleanup untile lowering to remove platform dependence.
Authored October 19, 2023
[random] deprecate jax.random.threefry_2x32 & threefry2x32_p
Authored October 18, 2023
AOT sharding mismatch error shouldn't have GSPMDSharding in it.
Authored October 25, 2023
[XLA:Mosaic] Support shrinking lane dim to 128 in shapecast.
Authored October 19, 2023
Invalidate in-memory caches on XLA-AutoFDO profile version change.
Authored October 27, 2023
Add mitigation techniques in the error message when a barrier timeout occurs.
Authored October 27, 2023
Delete cached_call_jaxpr_lowerings since a more general cached_primitive_lowerings is available
Authored October 30, 2023
Update the TPU dialect binding extension to follow MLIR guidelines
Authored October 24, 2023
Deprecated accessing `config` via the `jax.config` submodule
Authored October 27, 2023
[CI] avoid referencing numpy.core to fix nightly CI
Authored October 23, 2023
Set public module for exported jax.dtypes APIs
Authored October 17, 2023
Relax some test tolerances for TPU.
Authored October 24, 2023
CI: update pre-commit hooks to latest version
Authored October 30, 2023
[Mosaic] apply_vector_layout C++ rewrite: Fix updateSlice[FromRange] for empty slices
Authored October 19, 2023
[random] cleanup internal implementation
Authored October 17, 2023

Browse Other Releases

Latest Pending Unreleased 😎
jax-v0.4.22-rc Released December 13, 2023
7,434 Ξ”
jax-v0.4.21 Released December 4, 2023
0 Ξ”
jax-v0.4.21-rc Released December 4, 2023
15,642 Ξ”
jax-v0.4.20 Released November 2, 2023
0 Ξ”
jax-v0.4.20-rc Released November 2, 2023
5,123 Ξ”
jax-v0.4.19 Released October 18, 2023
0 Ξ”
jax-v0.4.19-rc Released October 18, 2023
5,722 Ξ”
jax-v0.4.18 Released October 6, 2023
0 Ξ”
jax-v0.4.18-rc Released October 6, 2023
3,149 Ξ”
jax-v0.4.17 Released October 3, 2023
0 Ξ”

Top Contributors in jax-v0.4.20-rc

mattjj
jakevdp
gnecula
superbobry
apaszke
yashk2810
sharadmv
froystig
hawkinsp
a-googler

Directory Browser for jax-v0.4.20-rc

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