Home » Oreps » Jax » All Releases » 0.3.24 Release
google/jax
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
Watch
0
Commit Activity
Directory Browser
Velocity History
Releases0.3.24
Basic Details Issues Addressed Top Contributors Directory Browser
Jax 0.3.24

Jax: 0.3.24 Release

Release date:
November 4, 2022
Previous version:
0.3.23 (released October 12, 2022)
Magnitude:
23,084 Diff Delta
Contributors:
37 total committers
Data confidence:
Commits:
181 impacting commits

181 Commits in this Release

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

Migrate remaining tests from jtu.cases_from_list to jtu.sample_product.
Authored October 12, 2022
Merge CUDA and ROCM kernel code in jaxlib.
Authored October 25, 2022
[typing] annotate next section of lax_numpy.py
Authored October 27, 2022
[typing] annotage jax.scipy.signal
Authored October 12, 2022
[typing] annotate jax.numpy ufuncs
Authored October 20, 2022
Enable state effect in `cond_p` (except in `grad` and `vmap`)
Authored November 2, 2022
Add an experimental custom_partitioner API which allows
Authored October 14, 2022
[typing] annotate jax.numpy array creation routines
Authored October 18, 2022
[typing] add type annotations to the first several lax_numpy functions
Authored October 21, 2022
[typing] annotate next chunk of lax_numpy.py
Authored October 25, 2022
Avoid loading scipy eagerly.
Authored October 6, 2022
Remove error-prone most_recent_entry() support from lu.cache.
Authored October 27, 2022
Remove JAX custom call implementation of batched triangular solve.
Authored October 18, 2022
improve (and shorten!) pmap error messages about inconsistent axis sizes
Authored October 13, 2022
Add Github Actions workflow that runs on a self-hosted TPU VM runner.
Authored October 21, 2022
Call _check_arraylike for jnp.linalg & jnp.fft functions
Authored October 27, 2022
Add new config `jax_persistent_cache_min_compile_time_secs`.
Authored October 28, 2022
[typing] annotate jnp.pad
Authored October 26, 2022
jax.Array is a unified type that will subsume JAX's DeviceArray, ShardedDeviceArray and GlobalDeviceArray.
Authored October 14, 2022
Rollback because of failing tests internally.
Authored October 14, 2022
[typing] annotate jax._src.dtypes
Authored October 12, 2022
Add JAX equivalent of scipy.stats.mode
Authored October 6, 2022
Add new config `jax_persistent_cache_min_instruction_count`.
Authored October 13, 2022
Add `lax.cumlogsumexp` for cumulative logsumexp operations.
Authored October 31, 2022
initial commit of DevicesSharding (fka SimpleSharding)
Authored October 15, 2022
Cleanup: remove unused imports in private modules
Authored October 20, 2022
implement truncnorm in jax.scipy.stats
Authored October 3, 2022
[jax2tf] Uses MHLO bytecode for XlaCallModule op.
Authored October 27, 2022
make leak checker errors explain why objects are alive
Authored October 26, 2022
Move jaxlib cpu kernels under jaxlib/cpu/.
Authored October 24, 2022
Annotate tree_util
Authored January 3, 2022
partitionable threefry PRNG random bits implementation
Authored October 25, 2022
Enable more GPU and TPU tests that pass at head.
Authored October 19, 2022
TMP: annotate util.safe_zip
Authored October 19, 2022
Use `d.process_index == d.client.process_index()` to find addressable devices from a list of addressable and non-addressable devices.
Authored October 28, 2022
[JAX:CPU] Enable buffer donation on CPU.
Authored October 26, 2022
[typing] annotate jax._src.numpy.index_tricks
Authored October 18, 2022
Add colors to sharding visualization
Authored November 1, 2022
[dynamic-shapes] fix nested vmap callable annotation logic
Authored October 17, 2022
[sparse] Make BCSR vmappable.
Authored October 14, 2022

Browse Other Releases

Latest Pending Unreleased 😎
0.3.25 Released November 15, 2022
6,881 Ξ”
0.3.24 Released November 4, 2022
23,084 Ξ”
0.3.23 Released October 12, 2022
305 Ξ”
0.3.22 Released October 11, 2022
11,644 Ξ”
0.3.21 Released September 30, 2022
1,561 Ξ”
0.3.20 Released March 9, 2023
101,633 Ξ”
jaxlib-v0.4.6 Released March 9, 2023
17,010 Ξ”
jaxlib-v0.4.4 Released February 16, 2023
6,724 Ξ”
jaxlib-v0.4.3 Released February 8, 2023
7,645 Ξ”
jaxlib-v0.4.2 Released January 25, 2023
24,737 Ξ”

Top Contributors in 0.3.24

hawkinsp
jake-vanderplas-1c1d
yashk2810
mattjj
pschuh
skye
sharadmv
gnecula
a-googler
tlu7

Directory Browser for 0.3.24

All files are compared to previous version, 0.3.23. Click here to browse diffs between other versions.

Loading File Browser...
GitClear uses cookies to ensure you get the best experience on our website. Learn more