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

Jax: jax-v0.4.11-rc Release

Release date:
May 31, 2023
Previous version:
jax-v0.4.10 (released May 11, 2023)
Magnitude:
6,799 Diff Delta
Contributors:
28 total committers
Data confidence:
Commits:
129 impacting commits

129 Commits in this Release

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

Move triton_kernel_call_lib to jaxlib
Authored May 24, 2023
Add (optional) ordered effects for `jax2tf.call_tf`
Authored May 24, 2023
remove `physical_avals` rule in favor of `physical_element_aval`
Authored May 11, 2023
allow lax.dot_general to accept different input dtypes
Authored May 13, 2023
Allow pjit.AUTO to be used with jax.jit. This introduces an API change which requires a mesh to be provided to pjit.AUTO(mesh).
Authored May 21, 2023
Add some preliminary support for int4/uint4 types to JAX.
Authored May 18, 2023
Add building on JAX
Authored May 22, 2023
Add new `called_index` to custom_call `tf.backend_config` DictAttr.
Authored May 25, 2023
Simplify custom_partitioning to use jax.ShapeDtypeStruct instead of passing separate
Authored May 18, 2023
inline and remove `eq_mlir` and `ne_mlir` rules
Authored May 12, 2023
PRNGKeyArray: add several missing attributes & methods
Authored May 17, 2023
[shape_poly] Add partial support for call_exported with polymorphic shapes
Authored April 26, 2023
Make pjit_call_impl go via C++ dispatch.
Authored May 26, 2023
Custom PRNG: support lax.full() and related constructors
Authored May 17, 2023
inline and remove `empty_mlir` rules
Authored May 12, 2023
inline and remove `transpose_mlir` rules
Authored May 12, 2023
explicitly convert jnp.var scalar normalizer to float (from int)
Authored May 23, 2023
Extend plugin discovery to also include entry-points.
Authored May 19, 2023
Add impl rule for with_sharding_constraint so that users can use their functions with and without a jit.
Authored May 17, 2023
Fix jax2tf_test regression failure.
Authored May 24, 2023
Gensym comparator names when lowering approx_top_k via MLIR
Authored May 16, 2023
Switch to using JAX status macros in jax-triton kernel call lib.
Authored May 25, 2023
Fix global_array_to_host_local_array when the specified pspec and mesh do not match the sharding of the input array.
Authored May 12, 2023
inline and remove `select_mlir` rules
Authored May 12, 2023
Checkify: Fix closing over Tracer in while_loop cond_f.
Authored May 17, 2023
inline and remove `dynamic_update_slice_mlir` rules
Authored May 12, 2023
Custom PRNG: make KeyArray compatible with custom_jvp
Authored May 17, 2023
Use new matrix_transpose in linalg code
Authored May 25, 2023
outline jitted `jax.random` functions
Authored May 25, 2023
Disable tests that time out in CI.
Authored May 17, 2023
inline and remove `dynamic_slice_mlir` rules
Authored May 12, 2023
Update comments
Authored May 19, 2023
Custom PRNG: support PRNGKeyArray.copy()
Authored May 12, 2023
inline and remove `broadcast_in_dim_mlir` rules
Authored May 12, 2023
Second attempt at fixing warnings from jax.dtypes.issubdtype.
Authored May 17, 2023
Break cycle between _src/core.py and _src/dtypes.py.
Authored May 17, 2023
Allow disabling the parsing of GSPMDSharding -> NamedSharding.
Authored May 24, 2023
custom PRNG: better error when using key as seed
Authored May 16, 2023
Fix sphinx syntax error.
Authored May 5, 2023
Remove axis_resources from with_sharding_constraint since it has been 3 months since the deprecation as per the API deprecation policy.
Authored May 26, 2023

Browse Other Releases

Latest Pending Unreleased 😎
jax-v0.4.13-rc Released June 22, 2023
7,868 Ξ”
jax-v0.4.12 Released June 8, 2023
0 Ξ”
jax-v0.4.12-rc Released June 8, 2023
4,680 Ξ”
jax-v0.4.11 Released May 31, 2023
0 Ξ”
jax-v0.4.11-rc Released May 31, 2023
6,799 Ξ”
jax-v0.4.10 Released May 11, 2023
0 Ξ”
jax-v0.4.10-rc Released May 11, 2023
1,233 Ξ”
jax-v0.4.9 Released May 9, 2023
1,058 Ξ”
jax-v0.4.9-rc Released May 9, 2023
23,579 Ξ”
jax-v0.4.8 Released March 29, 2023
2,231 Ξ”

Top Contributors in jax-v0.4.11-rc

froystig
yashk2810
jakevdp
sharadmv
hawkinsp
a-googler
mattjj
maxwillzq
pschuh
canyon289

Directory Browser for jax-v0.4.11-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