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

Jax: jax-v0.3.15-rc Release

Release date:
July 22, 2022
Previous version:
jax-v0.3.14 (released June 27, 2022)
Magnitude:
18,821 Diff Delta
Contributors:
36 total committers
Data confidence:
Commits:
182 impacting commits

182 Commits in this Release

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

Update minimum jaxlib version to 0.3.14.
Authored July 8, 2022
Add initial LOBPCG top-k eigenvalue solver (#3112)
Authored June 2, 2022
Convert everything in pjit to the `Sharding` interface. The following contains the things that have changed in this CL:
Authored July 7, 2022
Enable Python callbacks on TFRT TPU backend
Authored July 7, 2022
[dynamic shapes] revive iree
Authored June 29, 2022
[dynamic-shapes] revive basic bounded int machinery, add tests
Authored June 29, 2022
jax.mask and jax.shapecheck are being deprecated
Authored July 21, 2022
Implementation of jax.scipy.stats.gaussian_kde
Authored June 23, 2022
* Remove AUTO from MeshPspecSharding and treat it like _UNSPECIFIED singleton value.
Authored July 8, 2022
[jax2tf] A new experimental version with JAX native lowering.
Authored April 12, 2022
refactor `stages` types, adding methods for text and for cost/memory analyses
Authored July 2, 2022
Add JAX debugger
Authored June 28, 2022
Add `for_loop` primitive and impl rule
Authored June 22, 2022
[dynamic-shapes] Expand the handling of dynamic shapes for reshape and iota.
Authored November 16, 2021
Bump minimum jaxlib version to 0.3.10
Authored June 28, 2022
[dynamic-shapes] start basic vmap compatibility
Authored July 7, 2022
Fix mypy annotations
Authored June 27, 2022
[jax2tf] Add support for common audio convolutions (1D variants, dilated depthwise, transpose with SAME padding).
Authored June 30, 2022
Enable debugging callbacks with pjit on TPU
Authored July 22, 2022
Add scan implementation using `for` and tests
Authored June 28, 2022
Document how to run tests using Bazel.
Authored July 3, 2022
[jax2tf] Fixes for handling of convolutions with shape_polymorphism and enable_xla=False
Authored July 12, 2022
Change the internals of `with_sharding_constraint` to use the sharding instances.
Authored July 7, 2022
DOC: ensure that _wraps() generates correct links to wrapped functions
Authored July 21, 2022
Jax caches should depend on axis env.
Authored June 28, 2022
Replace `out_specs` with `out_shardings` and remove `out_indices` in ResultsHandler.
Authored July 19, 2022
Add partial_eval rule for `for`
Authored June 29, 2022
[shape-poly] Improve the error reporting for division
Authored July 13, 2022
* Check if the device assignment is the same across input and output shardings.
Authored July 11, 2022
Add a limiter for in-flight bytes. Read a shard from TensorStore if there are enough bytes are available. This only works for deserialization right now.
Authored July 2, 2022
Fix PE not allowing double JIT-ted effectful functions
Authored July 21, 2022
[jax2tf] Raise errors for experimental_native_lowering and custom_call
Authored July 21, 2022
[dynamic-shapes] add basic abstracted_axes support to jit(f, ...).lower(...)
Authored July 7, 2022
OpSharding doesn't have `__eq__` defined on it. Don't check sharding equality using opsharding until it does support that.
Authored July 20, 2022
Add caching to `trace_to_subjaxpr_dynamic2`.
Authored July 19, 2022
Add loop-invariant residual optimization for `for`
Authored July 8, 2022
Increase bazel sharding of GPU tests.
Authored July 11, 2022
Add support for post_process of xmap in BatchTrace
Authored July 5, 2022
Add tests for `jvp(for_loop)`
Authored June 28, 2022
add upstream-nightly CI job
Authored July 13, 2022

Browse Other Releases

Latest Pending Unreleased 😎
jax-v0.3.18-rc Released September 26, 2022
17,611 Ξ”
jax-v0.3.17 Released August 31, 2022
20,878 Ξ”
jax-v0.3.16 Released August 11, 2022
16,661 Ξ”
jax-v0.3.15 Released July 22, 2022
0 Ξ”
jax-v0.3.15-rc Released July 22, 2022
18,821 Ξ”
jax-v0.3.14 Released June 27, 2022
25,712 Ξ”
jax-v0.3.13 Released May 16, 2022
190 Ξ”
jax-v0.3.12 Released May 16, 2022
13 Ξ”
jax-v0.3.11 Released May 14, 2022
9,163 Ξ”
jax-v0.3.10 Released May 4, 2022
1,645 Ξ”

Top Contributors in jax-v0.3.15-rc

sharadmv
gnecula
yashk2810
mattjj
hawkinsp
jakevdp
vlad17
dfm
froystig
a-googler

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