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

Jax: jaxlib-v0.4.18 Release

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

64 Commits in this Release

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

[Mosaic] apply_vector_layout C++ rewrite: Have each rule use its own builder instead of passing a builder in the context
Authored October 5, 2023
sparse_test: Split into two so that each target is small enough to fit within a medium timeout
Authored October 5, 2023
[Mosaic] apply_vector_layout C++ rewrite: generalization for relayouts that reduce sublanes
Authored October 4, 2023
[Mosaic] apply_vector_layout C++ rewrite (14): tpu.matmul and vector.contract
Authored October 5, 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
identify PRNG schemes on key arrays, and recognize them in key constructors
Authored October 4, 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
[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
[export] Add tests for multi-platform and cross-platform export
Authored October 4, 2023
[run_state] add pjit run_state discharge rule and basic test
Authored October 4, 2023
[Pallas] Add pretty printing for dma_start
Authored October 4, 2023
Improve CUDA install documentation
Authored October 6, 2023
random_test: Split into two so that each target is small enough to fit within a medium timeout
Authored October 5, 2023
Allow duration event listeners to take extra keyword arguments
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
[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
Fix tests that mistakenly used assertRaises(..., msg=...) to match a message
Authored October 6, 2023
[run_state] add scan nested test, tweak rule name to mention 'state'
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
Check dtypes in fft_p's abstract eval rule
Authored October 6, 2023
jax.lax: deprecate inadvertent exports & internal utilities
Authored October 6, 2023
[pallas] Fix `allow_tf32` value in Triton `dot_general` lowering
Authored October 5, 2023
test custom PRNG impl construction round trip
Authored October 6, 2023
[Mosaic] Allow arbitrary LHS tensor axis size for non-packed matmuls
Authored October 4, 2023
Simplify the torch data loader collate function using tree_map
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
[Mosaic] apply_vector_layout: Use shape in generalizes check (in Python)
Authored October 5, 2023
[Mosaic] Run verifier after infer_memref_layout
Authored October 4, 2023

Browse Other Releases

Latest Pending Unreleased 😎
jaxlib-v0.4.22 Released December 13, 2023
6,291 Ξ”
jaxlib-v0.4.21 Released December 4, 2023
18,528 Ξ”
jaxlib-v0.4.20 Released November 2, 2023
4,475 Ξ”
jaxlib-v0.4.19 Released October 18, 2023
7,111 Ξ”
jaxlib-v0.4.18 Released October 6, 2023
4,220 Ξ”
jaxlib-v0.4.17 Released October 3, 2023
9,022 Ξ”
jaxlib-v0.4.16 Released September 18, 2023
9,933 Ξ”
jaxlib-v0.4.15 Released August 30, 2023
15,453 Ξ”
jaxlib-v0.4.14 Released July 27, 2023
17,116 Ξ”
jaxlib-v0.4.13 Released June 22, 2023
6,666 Ξ”

Top Contributors in jaxlib-v0.4.18

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

Directory Browser for jaxlib-v0.4.18

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 about our cookie policy