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

Jax: jax-v0.4.18-rc Release

Release date:
October 6, 2023
Previous version:
jax-v0.4.17 (released October 3, 2023)
Magnitude:
3,149 Diff Delta
Contributors:
16 total committers
Data confidence:
Commits:
61 impacting commits

61 Commits in this Release

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

[Mosaic] apply_vector_layout C++ rewrite (14): tpu.matmul and vector.contract
Authored October 5, 2023
identify PRNG schemes on key arrays, and recognize them in key constructors
Authored October 4, 2023
[Mosaic] apply_vector_layout C++ rewrite (16): vector.multi_reduction
Authored October 5, 2023
[Mosaic] infer_memref_layout C++ rewrite
Authored October 5, 2023
`_StateContextManager` now preserves the type of the value it stores.
Authored October 4, 2023
Add support for tiled layouts in arith.extf + implement (2,128)->(8,128) relayout for 32-bit types
Authored October 6, 2023
[Mosaic] apply_vector_layout C++ rewrite (15): vector.shape_cast
Authored October 5, 2023
[Mosaic] apply_vector_layout C++ rewrite (13): scf.if, scf.yield
Authored October 4, 2023
[export] Add tests for multi-platform and cross-platform export
Authored October 4, 2023
[JAX] Keep CPU host callbacks alive via IFRT, rather than by attaching them to the Python object.
Authored October 5, 2023
[Mosaic] Fix a buggy vector.broadcast rule in apply_vector_layout
Authored October 5, 2023
[Mosaic] apply_vector_layout C++ rewrite: various bug fixes
Authored October 5, 2023
random_test: Split into two so that each target is small enough to fit within a medium timeout.
Authored October 5, 2023
Improve CUDA install documentation.
Authored October 6, 2023
[run_state] add pjit run_state discharge rule and basic test
Authored October 4, 2023
Remove references to the ROCm TensorFlow repository in AMD build instructions.
Authored October 4, 2023
random_test: remove unnecessary test utilities
Authored October 5, 2023
Fix tests that mistakenly used assertRaises(..., msg=...) to match a message.
Authored October 6, 2023
[Pallas] Add pretty printing for dma_start
Authored October 4, 2023
[run_state] add scan nested test, tweak rule name to mention 'state'
Authored October 4, 2023
Allow duration event listeners to take extra keyword arguments.
Authored October 4, 2023
Add jax.numpy.bitwise_count()
Authored October 3, 2023
jax.random: fix NaN corner-case in loggamma
Authored October 4, 2023
[XLA:Mosaic] Fix layout attribute string parse in python.
Authored October 5, 2023
Clarify that mesh_utils.create_device_mesh's contiguous_submeshes arg isn't necessary with jax.Array
Authored October 4, 2023
Check dtypes in fft_p's abstract eval rule.
Authored October 6, 2023
jax.lax: deprecate inadvertent exports & internal utilities
Authored October 6, 2023
test custom PRNG impl construction round trip
Authored October 6, 2023
Simplify the torch data loader collate function using tree_map.
Authored October 4, 2023
[pallas] Fix `allow_tf32` value in Triton `dot_general` lowering.
Authored October 5, 2023
[Mosaic] Allow arbitrary LHS tensor axis size for non-packed matmuls
Authored October 4, 2023
[JAX] Obtain NCCL via a stub, rather than linking it statically or dynamically.
Authored October 4, 2023
lax.abs: better error for unsigned inputs
Authored October 5, 2023
[Mosaic] apply_vector_layout C++ rewrite (17): for.op (from cl/568376871)
Authored October 5, 2023
[export] Fix dot_general multi-platform lowering
Authored October 6, 2023
Update setup.py and CHANGELOG for jax 0.4.18 release
Authored October 6, 2023
Move the prerequisites to the beginning of the tests instructions.
Authored October 2, 2023
Update versions and CHANGELOG after jax 0.4.17 release
Authored October 4, 2023
Re-add JIT to jax.numpy.bitwise_count
Authored October 5, 2023
[Mosaic] apply_vector_layout: Use shape in generalizes check (in Python)
Authored October 5, 2023

Browse Other Releases

Latest Pending Unreleased 😎
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 Ξ”
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 Ξ”

Top Contributors in jax-v0.4.18-rc

tlongeri
hawkinsp
froystig
apaszke
jakevdp
superbobry
a-googler
mattjj
gnecula
skye

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