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

Jax: jax-v0.4.17-rc Release

Release date:
October 3, 2023
Previous version:
jax-v0.4.16 (released September 18, 2023)
Magnitude:
8,222 Diff Delta
Contributors:
30 total committers
Data confidence:
Commits:
173 impacting commits

173 Commits in this Release

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

[Mosaic] apply_vector_layout C++ rewrite (5) vector load and store
Authored September 20, 2023
MAINT Clean up leftover `Array = Any` aliases in jax/_src/**.py
Authored September 25, 2023
[CUDA] Add code to jax initialization that verifies that the CUDA libraries that are found are at least as new as the versions against which JAX was built.
Authored September 27, 2023
[jax_export] Add the next part of multi-platform lowering support.
Authored September 4, 2023
[Pallas] Automatically turn mesh indices -> physical ids for remote DMAs
Authored October 3, 2023
Better errors for array scalar/boolean conversion
Authored September 19, 2023
[typing] better annotations for jnp.ufunc
Authored September 20, 2023
Add a new internal test utility test_device_matches() and use it instead of equality tests on device_under_test().
Authored September 27, 2023
[Mosaic] Add `sin` and `clamp` lowering rules and support multiple branches in `cond`. Add a pallas_call test using scan/cond. Improve the error message for lowering exceptions and add a `LoweringException` type.
Authored September 27, 2023
Add dab/causal handling to flash attention backward pass
Authored September 19, 2023
[jax2tf] Fix higher-order differentiation.
Authored September 19, 2023
[Mosaic] apply_vector_layout C++ rewrite (11): vector.broadcast
Authored September 28, 2023
[Mosaic] apply_vector_layout C++ rewrite (8): tpu.gather, tpu.iota, tpu.trace
Authored September 28, 2023
Improve installation instructions.
Authored September 22, 2023
Relax shape constraints on matmul unrolling
Authored September 25, 2023
Introduce a LoweringParameters dataclass for easier plumbing
Authored September 28, 2023
Remove the skip_on_xla_cpu_mlir decorator.
Authored September 25, 2023
[Mosaic] Introduce tile_strides to memref layout to support slice.
Authored September 23, 2023
Add more API set up for Mock GPU client. Also clean up previous mock GPU client
Authored October 2, 2023
Clean up some device opt-in/opt-outs in test suite.
Authored September 27, 2023
Refer to CUDA stubs directly from TSL, rather than using an alias defined in xla/stream_executor.
Authored September 20, 2023
issubdtype: fix corner cases with extended dtypes
Authored September 22, 2023
Improve behavior of core.valid_jaxtype
Authored September 22, 2023
Fix typos across the package
Authored September 22, 2023
MAINT Use a generator expression in tuple([... for ... in ...])
Authored September 21, 2023
[random] add itemsize property to custom PRNG
Authored September 25, 2023
Fix test breakage in RNN test with old jaxlibs.
Authored September 20, 2023
[Mosaic] apply_vector_layout C++ rewrite (10): vector.extract_strided_slice
Authored September 28, 2023
Run LSTM test using FP32 math (as opposed to TF32)
Authored September 18, 2023
[Mosaic] apply_vector_layout C++ rewrite (9): tpu.repeat
Authored September 28, 2023
If an input to `jnp.asarray` is a numpy array, then convert it to a jax.Array via device_put to avoid a copy.
Authored September 22, 2023
Remove code that conditionally disables fp8 and int4 types in dtypes_test.
Authored September 26, 2023
dtypes.issubdtype: validate a when b is dtypes.extended
Authored September 21, 2023
Add cuda 12.2 support to JAX
Authored September 25, 2023
move `wrap_key_data` to `jax.random`
Authored September 18, 2023
[random] deprecate named key creation functions
Authored September 21, 2023
Add notes about the new CUDA version restrictions to the changelog and installation instructions.
Authored September 27, 2023
[Mosaic] apply_vector_layout C++ rewrite (7): tpu load and store
Authored September 25, 2023
Remove test code that checks for the se_tpu runtime.
Authored September 25, 2023
[shape_poly] Ensure we can have both ordered effects and dead variables
Authored September 21, 2023

Browse Other Releases

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

Top Contributors in jax-v0.4.17-rc

hawkinsp
jakevdp
tlongeri
gnecula
superbobry
sharadmv
a-googler
emilyfertig
yashk2810
bythew3i

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