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

Jax: jax-v0.4.16-rc Release

Release date:
September 18, 2023
Previous version:
jax-v0.4.15 (released August 30, 2023)
Magnitude:
9,916 Diff Delta
Contributors:
32 total committers
Data confidence:
Commits:
149 impacting commits

149 Commits in this Release

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

Move memories_test.py to JAX
Authored September 12, 2023
[MOSAIC] apply_vector_layout C++ rewrite (3) applyLayoutOp and relayout
Authored September 7, 2023
Remove code that disabled tests on "stream_executor" backends.
Authored September 14, 2023
[PJRT C API] Set up jax xla cuda package.
Authored September 13, 2023
[export] Move jax_export and shape_poly out of jax2tf.
Authored September 6, 2023
[callback] Some test cleanup.
Authored September 3, 2023
rolling forward shard_map transpose fixes
Authored September 1, 2023
[JAX] Change jax.test_util utilities to have identical tolerances on all platforms.
Authored September 6, 2023
[Pallas] Add support for scoped allocations to Pallas TPU
Authored September 7, 2023
[export] Implement the calling convention for exporting with multi-platform lowering
Authored August 27, 2023
Support relayout of tiles in register when the layout tiling changes.
Authored September 7, 2023
fast dispatch for functions over typed PRNG key arrays
Authored September 13, 2023
Test changes for out-of-tree backend.
Authored September 13, 2023
[Pallas] Add support for allocating TPU semaphores and basic semaphore ops
Authored September 8, 2023
build: write appropriate version strings to build artifacts
Authored September 7, 2023
Update internal callers to avoid PRNGKeyArray
Authored September 13, 2023
Make `is_fully_addressable` an abstract method and implement it on each concrete Sharding.
Authored September 6, 2023
JEX: move jax.linear_util to jax.extend.linear_util
Authored August 30, 2023
deprecate `PRNGKeyArray.unsafe_raw_array` in favor of `jax.random.key_data`
Authored September 13, 2023
Set the jax_enable_memories flag to True.
Authored September 11, 2023
Hash serialized CompileOptions for new cache key generation.
Authored September 1, 2023
[Mosaic] Add a pass for instantiating memory spaces
Authored September 12, 2023
[Pallas] Add `async_copy_to` and `async_remote_copy_to` for doing DMAs.
Authored September 14, 2023
Add memories support to remat.
Authored September 13, 2023
[Mosaic] Add a tpu.matmul op
Authored September 5, 2023
Propagate ad.Zeros to the scan body function for jax.lax.scan for the output 'ys'.
Authored September 7, 2023
Internal: add dtypes.safe_to_cast utility & use to generate indexing warning
Authored September 7, 2023
Deprecate the contents of jax.prng
Authored August 30, 2023
Add more extensive tests for version strings
Authored September 8, 2023
[Mosaic] Add debug flag to print module after each path.
Authored August 31, 2023
[Pallas] Add support for local DMAs on TPU backend
Authored September 13, 2023
[callbacks] Add support for shardable ordered effects.
Authored September 18, 2023
[Pallas] Add support for remote DMAs on TPU
Authored September 13, 2023
Fully unroll the `scan` in `jnp.searchsorted`, when method 'scan_unrolled' is specified. On GPU, XLA's 'scan' (fori_loop) implementation launches multiple calls to the body_fun GPU kernel, whereas a fully unrolled scan can be fused into a single kernel launch.
Authored September 14, 2023
[MOSAIC] apply_vector_layout C++ rewrite (2) No-op pass and flag to use it instead of Python
Authored August 31, 2023
Make apply_primitive go via C++ fast dispatch.
Authored August 31, 2023
jax.numpy: add preferred_element_type argument to matmul functions
Authored September 7, 2023
Improve the gpu lowering error message if users forget link the gpu library.
Authored September 11, 2023
Add support for float flags to compiler_options.
Authored September 14, 2023
Remove references to deprecated "tpu_se" build configuration.
Authored September 13, 2023

Browse Other Releases

Latest Pending Unreleased 😎
jax-v0.4.18-rc Released October 6, 2023
3,149 Ξ”
jax-v0.4.17 Released October 3, 2023
0 Ξ”
jax-v0.4.17-rc Released October 3, 2023
8,498 Ξ”
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 Ξ”

Top Contributors in jax-v0.4.16-rc

yashk2810
gnecula
hawkinsp
jakevdp
sharadmv
tlongeri
jyingl3
a-googler
apaszke
mattjj

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