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

Jax: jax-v0.2.11 Release

Release date:
March 24, 2021
Previous version:
jax-v0.2.10 (released March 5, 2021)
Magnitude:
5,613 Diff Delta
Contributors:
21 total committers
Data confidence:
Commits:
89 impacting commits

89 Commits in this Release

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

Initial commit of seventh JAX-101 notebook
Authored March 8, 2021
Remove the JAX lazy sublanguage.
Authored March 10, 2021
[jax2tf] Added instructions for using OSS TensorFlow model server.
Authored March 17, 2021
[jax2tf] Removed more traces of support for batch polymorphism
Authored March 16, 2021
Increase minimum jaxlib version to 0.1.62.
Authored March 16, 2021
Make all_to_all primitive match XLA semantics
Authored March 5, 2021
Add scan and while rule for jax.experimental.callback transformation
Authored March 9, 2021
BUG: fix jnp.result_type for non-canonical weak types
Authored March 15, 2021
Substitute axis names in nested jaxprs
Authored March 8, 2021
Make jax_debug_nans and jax_debug_infs work with pmap, xmap, and pjit.
Authored March 12, 2021
simplify standard named_shape_rule
Authored February 10, 2021
Reenable multi-axis all_to_all
Authored March 5, 2021
Add new lax.rng_bit_generator primitive
Authored February 16, 2021
Initial implementation of jnp.delete
Authored March 17, 2021
Added 'where' keyword to 'jnp.{mean, var, std}'
Authored March 6, 2021
add a memory leak test for jit jaxpr construction
Authored March 17, 2021
Cleanups for laziness. No functional changes intended.
Authored March 5, 2021
jnp.array.transpose: support positional axis arguments
Authored March 11, 2021
Make lax._const() work for non-canonical dtypes
Authored March 17, 2021
Add pytree support to pjit's with_sharding_constraint
Authored February 24, 2021
make axis_index bind respect dynamic traces
Authored March 9, 2021
convert_element_type: don't canonicalize old_dtype
Authored March 12, 2021
jnp.power: fix overflow case for x1=0
Authored March 9, 2021
Try to avoid transposes in jnp.einsum by considering both argument orders to dot_general.
Authored March 4, 2021
Fix URL in custom errors
Authored March 16, 2021
[JAX] Refactor handling of JIT interpreter state in jax_jit API.
Authored March 17, 2021
Add extras_require to setup.py
Authored March 16, 2021
Add jtu.count_device_put for tests to count device_put.
Authored March 9, 2021
Add workaround for SelectAndScatter padding bug on CPU and GPU.
Authored March 10, 2021
Fix incorrect handling of constants in xmap's partial_eval rule
Authored March 15, 2021
add test for issue #5900, once jaxlib is updated
Authored March 5, 2021
[XLA:Python] Split DevicePut out of `jax_jit` and refactor it.
Authored March 7, 2021
jax.random.poisson: fix return value for lam=0
Authored March 8, 2021
[JAX] Enable C++ device arrays by default.
Authored March 15, 2021
jnp.piecewise: support scalar inputs
Authored March 9, 2021
tweak broken tests
Authored March 10, 2021
Fix crash returning a Token from a jit computation on GPU.
Authored March 10, 2021
avoid modifying global pytype mappings in named-shape jaxpr test
Authored February 11, 2021
[host_callback] Enable configuring the outfeed receiver buffer size
Authored March 18, 2021
Automatically initialize Cloud TPU topology env vars if running on a Cloud TPU VM.
Authored March 5, 2021

Browse Other Releases

Latest Pending Unreleased 😎
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 Ξ”
jax-v0.2.10 Released March 5, 2021
0 Ξ”
jax-v0.2.9 Released January 27, 2021
0 Ξ”
jax-v0.2.8 Released January 12, 2021
0 Ξ”
jax-v0.2.6 Released November 18, 2020
0 Ξ”

Top Contributors in jax-v0.2.11

jakevdp
hawkinsp
gnecula
apaszke
skye
mattjj
froystig
sharadmv
tamas-berghammer-3afb
zhangqiaorjc

Directory Browser for jax-v0.2.11

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