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

Jax: jaxlib-v0.3.15 Release

Release date:
July 22, 2022
Previous version:
jaxlib-v0.3.14 (released June 27, 2022)
Magnitude:
16,976 Diff Delta
Contributors:
34 total committers
Data confidence:
Commits:
161 impacting commits

161 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
* 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 `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
[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
Add dlpack support to device_array and jax.numpy
Authored July 15, 2022
Save an on-demand checkpoint when any worker receives a preemption signal.
Authored July 1, 2022
Create `AsyncManager`, which factors out thread management functionality from `GlobalAsyncCheckpointManager` and makes it available for use (such as in Orbax) by classes supporting async read/write.
Authored June 29, 2022

Browse Other Releases

Latest Pending Unreleased 😎
jaxlib-v0.3.25 Released November 15, 2022
7,635 Ξ”
jaxlib-v0.3.24 Released November 4, 2022
23,851 Ξ”
jaxlib-v0.3.22 Released October 11, 2022
12,900 Ξ”
jaxlib-v0.3.20 Released September 28, 2022
55,337 Ξ”
jaxlib-v0.3.15 Released July 22, 2022
16,976 Ξ”
jaxlib-v0.3.14 Released June 27, 2022
33,978 Ξ”
jaxlib-v0.3.10 Released May 4, 2022
13,335 Ξ”
jaxlib-v0.3.7 Released April 15, 2022
10,780 Ξ”
jaxlib-v0.3.5 Released April 7, 2022
9,385 Ξ”
jaxlib-v0.3.2 Released March 15, 2022
19,405 Ξ”

Top Contributors in jaxlib-v0.3.15

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

Directory Browser for jaxlib-v0.3.15

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