Home » Oreps » Jax » All Releases » jax-v0.2.19 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.19

Jax: jax-v0.2.19 Release

Release date:

August 13, 2021

Previous version:

jax-v0.2.18 (released July 21, 2021)

Magnitude:

12,452 Diff Delta

Contributors:

26 total committers

Data confidence:

Commits:

150

150 Commits in this Release

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

Use the new "custom call status" facility to report errors in jaxlib

Authored August 9, 2021

[jax2tf] Improved error checking for call_tf.

Authored July 10, 2021

[jax2tf] Refactored the handling of float0.

Authored July 27, 2021

test_util: add decorator to set config values in test cases

Authored August 5, 2021

Add Python code for the future C++ pmap and pass the data to C++ as a namedtuple.

Authored August 4, 2021

[jax2tf] Allows shape polymorphic specification to be polynomials.

Authored August 4, 2021

Remove Python 3.6 compatibility code.

Authored July 29, 2021

Create a .bazelrc file that is the base for all the builds. The current build.py workflow will not be affected since this .bazelrc file will be overridden. I am going to change that workflow in the coming CLs.

Authored August 10, 2021

DOC: many small fixes

Authored August 3, 2021

[jax2tf] Test code refactoring

Authored July 28, 2021

ReduceScatter translation and abstract eval.

Authored July 27, 2021

Make it possible to return a C++ `ShardedDeviceArray`.

Authored August 10, 2021

Add @jit decorators to another tranche of jax.numpy functions.

Authored August 12, 2021

small autodidax tweaks

Authored August 5, 2021

Ensure zeros from AD are generated on device.

Authored June 25, 2021

Expose UnexpectedTracerError and add docs.

Authored June 30, 2021

Improve implementation of cbrt() in JAX.

Authored July 22, 2021

small tweaks, type annotations

Authored July 23, 2021

Add `@jit` decorators to jax.numpy operators.

Authored August 10, 2021

Fix some of the formatting and reword some of the sections.

Authored July 28, 2021

Enable constructing C++ ShardedDeviceArray instead of the Python object.

Authored August 11, 2021

[host_callback] Fix the JVP rule for id_tap(result=...)

Authored July 25, 2021

[jax2tf] enable multiple round-trips from JAX to TF

Authored July 30, 2021

[sparse] handle pytree inputs in sparsify transform

Authored August 10, 2021

[jax2tf] Adjustments for the polymorphic_shapes specification.

Authored July 23, 2021

[jax2tf] Improve support for converting functions with kwargs

Authored August 4, 2021

Use MainTrace payload mechanism in experimental callback tracer.

Authored August 5, 2021

[jax2tf] Add shape polymorphism support for jnp.eye

Authored August 3, 2021

Validate shapes for boolean indices

Authored August 3, 2021

[sparse] correctly handle units in sparsify argspecs

Authored August 9, 2021

Don't try to use persistent compilation cache when running CPU computations.

Authored August 6, 2021

[jax2tf] More cleanup for shape polymorphism testing

Authored July 30, 2021

improve and add to closure_convert testing

Authored July 24, 2021

Add tracers to LeakChecker error, and filter out false positives this way.

Authored July 21, 2021

[jax2tf] Update the limitations

Authored July 31, 2021

[sparse] add sparsify rule for lax.cond

Authored August 6, 2021

[jax2tf] Add support for the new XlaRngBitGenerator TensorFlow op

Authored August 6, 2021

[jax2tf] Improved testing and shape polymorphism support for lax.random.

Authored July 29, 2021

refine constant-hoisting heuristic for closure_convert

Authored July 22, 2021

Log a warning after 60 secs to remind the user to run code on all hosts for Cloud TPU 1VM

Authored July 28, 2021

Browse Other Releases

Latest Pending Unreleased 😎

jax-v0.2.24 Released October 19, 2021

298 Δ

jax-v0.2.23 Released October 19, 2021

0 Δ

jax-v0.2.22 Released October 13, 2021

37 Δ

jax-v0.2.21 Released September 23, 2021

10,380 Δ

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 Δ

Top Contributors in jax-v0.2.19

gnecula

agrue

jakevdp

hawkinsp

jblespiau

LenaMartens

yashk2810

mattjj

elliotwaite

xnning

Directory Browser for jax-v0.2.19

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