Home » Oreps » Jax » All Releases » jax-v0.2.15 Release

google/jax

Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more

Watch

Basic Details Issues Addressed Top Contributors Directory Browser

Jax jax-v0.2.15

Jax: jax-v0.2.15 Release

Release date:

June 23, 2021

Previous version:

jax-v0.2.14 (released June 10, 2021)

Magnitude:

6,137 Diff Delta

Contributors:

15 total committers

Data confidence:

Commits:

72 Commits in this Release

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

[jax2tf] Support inequality and min/max for booleans.

Authored June 10, 2021

[jax2tf] Fix the round-trip call_tf(convert)

Authored June 10, 2021

Implemented get_cache_key funtion with unit tests

Authored June 4, 2021

Add lax.linalg.tridiagonal_solve(..), lowering to cusparse_gtsv2<T>() on GPU.

Authored May 26, 2021

[sparse] add bcoo_transpose primitive

Authored June 23, 2021

[jax2tf] Add more documentation about saving models with custom gradients

Authored June 15, 2021

[call_tf] Fixed the handling of x64 values in JAX_ENABLE_X64=False

Authored June 16, 2021

[sparse] create experimental sparse submodule

Authored June 16, 2021

Add support for anonymous serial loops

Authored June 8, 2021

Drop support for NumPy 1.16.

Authored June 10, 2021

[jax2tf] Fix the custom gradients for call_tf

Authored June 16, 2021

[sparse] add sparse-aware wrapper for grad & value_and_grad

Authored June 18, 2021

Support complex numbers in jax.scipy.signal.convolve/correlate

Authored June 11, 2021

Improve concretization errors for jnp indexing routines

Authored June 17, 2021

Support value computation of associated Legendre functions.

Authored June 14, 2021

Add LRU eviction policy to persisent compilation cache

Authored June 2, 2021

Add auxiliary data support to lax.custom_root

Authored June 17, 2021

Implements more slicing support for lax.gather in jax2tf when enable_xla=False, and adds tests.

Authored June 22, 2021

Rename master to main in embedded links.

Authored June 18, 2021

[jax2tf] Improve dtype coverage for neg; update limitation documentation

Authored June 14, 2021

[jax2tf] Change the InconclusiveDimensionOperation error to include link to documentation

Authored June 13, 2021

add efficient path for array input to jnp.stack, jnp.[hvd]stack, jnp.column_stack

Authored June 11, 2021

[sparse] fix batched serialization of BCOO

Authored June 22, 2021

Implements dynamic_update_slice when enable_xla=False and adds tests.

Authored June 22, 2021

add bcoo_reduce_sum() function

Authored June 11, 2021

Fix random test for scipy>=1.17.0

Authored June 21, 2021

Add vmap rule for lax.clamp.

Authored June 21, 2021

Generalize BCOO.__matmul__

Authored June 14, 2021

[jax2tf] Add a best-effort conversion for gather with enable_xla=False

Authored June 14, 2021

Fix most test failures under NumPy 1.21.

Authored June 22, 2021

[jax2tf] Fix bug in gradient for functions with integer results

Authored June 15, 2021

[host_callback] Make logging statements conditional.

Authored June 21, 2021

[sparse] add bcoo_rdot_general

Authored June 16, 2021

[JAX] Improve support for DLPack tensors on CPU when a GPU is available.

Authored June 22, 2021

[sparse] add tests of sparse dot_general autodiff with respect to sparse argument

Authored June 18, 2021

[jax2tf] Extend shape polymorphism to handle add_transpose with broadcasting

Authored June 12, 2021

jnp.concatenate: add fast path based on lax.reshape for array inputs

Authored June 10, 2021

On Cloud TPU, use pip-installed libtpu instead of system default if applicable.

Authored June 22, 2021

random: generate large sequences via vmap rather than Python loops

Authored June 17, 2021

[sparse] update sparse object repr to show tracing

Authored June 15, 2021

Browse Other Releases

Latest Pending Unreleased 😎

jax-v0.2.20 Released September 3, 2021

0 Δ

jax-v0.2.19 Released August 13, 2021

12,452 Δ

jax-v0.2.18 Released July 21, 2021

0 Δ

jax-v0.2.17 Released July 9, 2021

6,049 Δ

jax-v0.2.16 Released June 23, 2021

61 Δ

jax-v0.2.15 Released June 23, 2021

6,137 Δ

jax-v0.2.14 Released June 10, 2021

0 Δ

jax-v0.2.13 Released May 4, 2021

1,558 Δ

jax-v0.2.12 Released April 1, 2021

0 Δ

jax-v0.2.11 Released March 24, 2021

5,611 Δ

Top Contributors in jax-v0.2.15

gnecula

jakevdp

colemanliyah

hawkinsp

tomhennigan

nicholasjng

marcvanzee

apaszke

lukepfister

tlu7

Directory Browser for jax-v0.2.15

We haven't yet finished calculating and confirming the files and directories changed in this release. Please check back soon.