Home » Oreps » Jax » All Releases » 0.3.20 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.20
Basic Details Issues Addressed Top Contributors Directory Browser
Jax 0.3.20

Jax: 0.3.20 Release

Release date:
March 9, 2023
Previous version:
jaxlib-v0.4.6 (released March 9, 2023)
Magnitude:
101,633 Diff Delta
Contributors:
119 total committers
Data confidence:
Commits:
1,272 impacting commits

1272 Commits in this Release

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

Merge CUDA and ROCM kernel code in jaxlib.
Authored October 25, 2022
Rewrite Checkify to support tracking different error types.
Authored October 18, 2022
(NFC) Prepare for migration from producing MHLO to producing StableHLO
Authored December 16, 2022
Refactor effects system to use effect types, not objects
Authored February 2, 2023
Add an internal jtu.sample_product test decorator.
Authored October 3, 2022
[jax2tf] Adds tests for polymorphic input for Flax Module Harnesses.
Authored January 10, 2023
[jax2tf] An alternative support for shape polymorphism for native serialization.
Authored November 28, 2022
Add CUDNN custom call for LSTM. Exposed as jax.experimental.rnn module.
Authored November 23, 2022
[jax2tf] Refactoring of shape_poly_test.
Authored December 17, 2022
[typing] annotate next section of lax_numpy.py
Authored October 27, 2022
add a basic prototype of piles, behind jax_dynamic_shapes
Authored October 17, 2022
[typing] add type annotations to jax.scipy.linalg
Authored October 10, 2022
Checkify: switch to initial-style.
Authored December 6, 2022
Replace jax._src.util.prod with math.prod.
Authored February 28, 2023
Copybara import of the project:
Authored October 9, 2022
Migrate api_test, lax_numpy_test, and lax_vmap_test to
Authored October 5, 2022
[shape_poly] Generalize binary operations with symbolic dimensions.
Authored January 18, 2023
Migrate JAX internals to builtin Python logging
Authored October 13, 2022
Refactor `Ref` abstract type to contain other `AbstractValue`s
Authored February 17, 2023
Delete `jax_experimental_name_stack` flag
Authored November 10, 2022
Switch to new-style f-strings
Authored December 1, 2022
[typing] annotage jax.scipy.signal
Authored October 12, 2022
[sparse] refactor/clean up test suite
Authored December 16, 2022
[sparse] improve testing framework
Authored November 16, 2022
[jax2tf] Allow the use of DimPolynomial with jnp.array and binary operations
Authored October 7, 2022
Roll back #14792
Authored March 8, 2023
Delete `jax.experimental.callback`
Authored January 13, 2023
Replace uses of jnp.ndarray with jax.Array inside JAX.
Authored February 15, 2023
[shape_poly, call_tf] Some improvements for call_tf in a shape polymorphic program
Authored March 1, 2023
[typing] annotate jax.numpy ufuncs
Authored October 20, 2022
migrate internal dependencies from `jax.interpreters.ad` to `jax._src.interpreters.ad`
Authored February 7, 2023
Switch lax_numpy_indexing_test to use jtu.sample_product.
Authored October 6, 2022
Multinodes CICD on GPUs using on-demand cluster and e2e tests using T5X
Authored September 26, 2022
Shorten alias chains for names exported in jax. namespace.
Authored March 1, 2023
Expose fp8 in jax dtypes and mlir builder.
Authored January 14, 2023
[shard-map] add systematic tests for eager, jit, autodiff
Authored February 15, 2023
DOC: add example of pure_callback with custom_jvp
Authored December 9, 2022
Add a number of missing function cross-references in the docs.
Authored August 24, 2022
Clean up: remove unused helper functions
Authored February 1, 2023
Enable state effect in `cond_p` (except in `grad` and `vmap`)
Authored November 2, 2022

Browse Other Releases

Latest Pending Unreleased 😎
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 Ξ”
jaxlib-v0.4.1 Released December 13, 2022
596 Ξ”

Top Contributors in 0.3.20

jake-vanderplas-1c1d
hawkinsp
yashk2810
gnecula
mattjj
sharadmv
pschuh
LenaMartens
zhangqiaorjc
burmako

Directory Browser for 0.3.20

All files are compared to previous version, jaxlib-v0.4.6. 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