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

Jax: jax-v0.4.15-rc Release

Release date:
August 30, 2023
Previous version:
jax-v0.4.14 (released July 27, 2023)
Magnitude:
14,550 Diff Delta
Contributors:
41 total committers
Data confidence:
Commits:
306 impacting commits

306 Commits in this Release

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

[callback] Disable stream_executor tests.
Authored August 22, 2023
Switch jaxlib to use nanobind instead of pybind11.
Authored August 24, 2023
[Memories] Add Memories support to jax.jit and jax.device_put!
Authored August 4, 2023
[MOSAIC] apply_vector_layout C++ rewrite (1) VectorLayout functions
Authored August 30, 2023
Improve the sharding documentation.
Authored August 3, 2023
Add an optional `sharding` argument to `pure_callback` and `io_callback`
Authored August 25, 2023
Add batching rules to jax.lax.linalg.tridiagonal_solve.
Authored August 10, 2023
Bump minimum_jaxlib_version to 0.4.14. `xla_extension_version` is 174 and `mlir_api_version` is 54
Authored August 1, 2023
[jax_export] Add backwards compatibility tests for shape_assertion.
Authored August 11, 2023
Add segment_ids support to pallas flash attention on GPU.
Authored August 29, 2023
jax.scipy.stats: add logsf & make sf more accurate near zero
Authored August 22, 2023
Add a type stub for jax.numpy.
Authored August 22, 2023
Remove the `canonicalize_dtypes` argument from mlir.ir_constant(s).
Authored August 17, 2023
Instead of tf.Graph protobuf, we switch to `tf.Saved_model` for back_compat_tf_test.
Authored August 10, 2023
Remove trivial execution from jax since it leads to 100x slower dispatch time.
Authored August 25, 2023
Remove allowlist for multihost collectives.
Authored August 9, 2023
Move functions out of xla.py closer to their users.
Authored August 8, 2023
Add an example of jax.tree_util.register_static to the jax 101 pytree docs.
Authored August 24, 2023
jnp.ufunc: support where argument in ufunc.reduce
Authored August 24, 2023
Replace references to DeviceArray with Array.
Authored August 18, 2023
Move compiler APIs out of dispatch.py and xla_bridge.py into a new jax._src.compiler module.
Authored August 15, 2023
Tracebacks no longer have JAX-internal frames prepended by default
Authored August 3, 2023
Add _manual_axes support to NamedSharding. This is needed because
Authored August 23, 2023
create `jax.extend.random` as a copy of `jax.prng`
Authored August 24, 2023
API: Remove NINF and PINF usages
Authored August 9, 2023
[JAX] Implement the `stream` argument to jax.Array.__dlpack__ for CUDA GPU
Authored August 18, 2023
Remove jax.interpreters.xla.register_collective_primitive.
Authored August 29, 2023
Make mlir.custom_call() more general and expose it as jax.interpreters.mlir.custom_call().
Authored August 29, 2023
jnp.ufunc: add fast paths for add/prod reductions
Authored August 28, 2023
Add `TransferToMemoryKind` as a private API to allow device_put to transfer to different memories without specifying the sharding and allowing the SPMD partitioner to choose the sharding for the intermediate.
Authored August 23, 2023
Deprecate jax.numpy.trapz.
Authored August 24, 2023
jnp.ufunc: add __hash__ method and jit methods by default
Authored August 14, 2023
[host_callback] Remove old backwards compatibility flag jax_host_callback_ad_transforms.
Authored August 16, 2023
Move some utilities out of dispatch.py next to their users, add more types.
Authored August 8, 2023
Improve dispatch.py typing.
Authored August 8, 2023
Create lax.polygamma with native HLO lowering
Authored August 16, 2023
add random.triangular and random.lognormal
Authored August 4, 2023
Create lax.zeta with native HLO lowering
Authored August 16, 2023
add an exp2 primitive and lax.exp2
Authored July 28, 2023
Implement jax.numpy.place with required inplace parameter
Authored August 2, 2023

Browse Other Releases

Latest Pending Unreleased 😎
jax-v0.4.17-rc Released October 3, 2023
8,222 Ξ”
jax-v0.4.16 Released September 18, 2023
0 Ξ”
jax-v0.4.16-rc1 Released September 18, 2023
3 Ξ”
jax-v0.4.16-rc Released September 18, 2023
9,916 Ξ”
jax-v0.4.15 Released August 30, 2023
0 Ξ”
jax-v0.4.15-rc Released August 30, 2023
14,550 Ξ”
jax-v0.4.14 Released July 27, 2023
0 Ξ”
jax-v0.4.14-rc Released July 27, 2023
19,891 Ξ”
jax-v0.4.13 Released June 22, 2023
0 Ξ”
jax-v0.4.13-rc Released June 22, 2023
7,868 Ξ”

Top Contributors in jax-v0.4.15-rc

hawkinsp
yashk2810
jakevdp
gnecula
a-googler
tlongeri
sharadmv
pschuh
junwhanahn
srvasude

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