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
1
Commit Activity
Directory Browser
Velocity History
Releasesjax-v0.2.15
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,130 Diff Delta
Contributors:
15 total committers
Data confidence:
Commits:
72 impacting 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,130 Ξ”
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,613 Ξ”

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.

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