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

Jax: jax-v0.2.28 Release

Release date:
February 2, 2022
Previous version:
jax-v0.2.27 (released January 18, 2022)
Magnitude:
7,023 Diff Delta
Contributors:
22 total committers
Data confidence:
Commits:
101 impacting commits

101 Commits in this Release

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

djax: let make_jaxpr build dyn shape jaxprs
Authored January 21, 2022
Implement SciPy's RegularGridInterpolator
Authored November 19, 2021
[JAX] Re-enables lowering bcoo dot general to cuSparse.
Authored January 31, 2022
disable implicit rank promotion in a number of remaining tests
Authored January 28, 2022
generate bit widths other than 32 in `lax.rng_bit_generator`
Authored January 21, 2022
linalg_test: disable implicit rank promotion
Authored January 26, 2022
checkify: fix and test post_process_call/map
Authored January 19, 2022
fix caching example in jit tutorial
Authored January 25, 2022
Make lapack symbols strong in lapack_kernels
Authored January 26, 2022
[XLA:TPU] Add 'aggregate_to_topk' option to ann in jax
Authored January 31, 2022
has_aux for jvp and forward-mode AD value_and_grad
Authored November 1, 2021
NotImplementedError for new_checkpoint-of-xmap
Authored January 20, 2022
jnp.[nan]argmin/max: implement keepdims
Authored January 20, 2022
Add GDA to the API pages
Authored January 21, 2022
Couple of changes because of the serialization inconsistencies being observed.
Authored February 1, 2022
[JAX] Change signature of .mhlo() method on compiler IR objects to return an ir.Module object instead of its string representation.
Authored January 19, 2022
custom vmap: abstract eval and translation rules
Authored January 15, 2022
Clarify the required NVidia driver versions in the JAX readme instructions.
Authored January 28, 2022
jaxpr staging: only one tracer per jaxpr variable
Authored January 25, 2022
Fix auto-generated docstrings for JIT-compiled functions
Authored January 21, 2022
introduce `custom_batching.sequential_vmap`
Authored January 16, 2022
[jax2tf] Move the no_xla_limitations documentation
Authored January 24, 2022
Simplify `gda_fast_path_args` by replacing local_idx_rid with global_indices_replica_id.
Authored January 20, 2022
Add the cache back now that Mesh's __hash__ is also being hashed on `self.devices.shape`.
Authored February 1, 2022
[sparse] avoid implicit rank promotion
Authored January 24, 2022
[sparse] add bcoo_sort_indices
Authored January 18, 2022
developer doc: more info on pre-commit
Authored January 25, 2022
re-enable tests of #8955
Authored January 20, 2022
Strip named_shape and weak_type from aval when donating buffers.
Authored January 28, 2022
[jax2tf] Improve shape polymorphism for scatter with CLIP mode
Authored January 20, 2022
Strip named_shape and weak_type from aval when donating buffers.
Authored January 28, 2022
jax.numpy: add where and initial arguments to nan reductions
Authored January 20, 2022
Copybara import of the project:
Authored January 21, 2022
jnp.angle: support deg keyword
Authored January 20, 2022
Cache the expensive computations in GDA. For example `get_shard_indices_replica_ids` can be the same for multiple variables in a neural network (global_shape, mesh_axes and global_mesh) can be the same
Authored January 18, 2022
[JAX:IREE] Fix bitrot in iree plugin for JAX.
Authored January 25, 2022
maint: update pre-commit package versions & fix new mypy errors
Authored January 31, 2022
Check the gda pjit mesh mismatch inside pjit and not during _shard_arg which is shared by pmap, xmap and pjit. For pmap, pjit mesh has nothing to do with it. So this error should not be raised.
Authored January 20, 2022
[JAX] Require exact type equality using `is` for static arguments.
Authored January 25, 2022
pjit: add test for basic static_argnums
Authored January 20, 2022

Browse Other Releases

Latest Pending Unreleased 😎
jax-v0.3.4 Released March 18, 2022
331 Ξ”
jax-v0.3.3 Released March 17, 2022
679 Ξ”
jax-v0.3.2 Released March 16, 2022
13,751 Ξ”
jax-v0.3.1 Released February 18, 2022
6,721 Ξ”
jax-v0.3.0 Released February 10, 2022
5,980 Ξ”
jax-v0.2.28 Released February 2, 2022
7,023 Ξ”
jax-v0.2.27 Released January 18, 2022
11,661 Ξ”
jax-v0.2.26 Released December 8, 2021
29,514 Ξ”
jax-v0.2.25 Released October 25, 2021
2,029 Ξ”
jax-v0.2.24 Released October 19, 2021
298 Ξ”

Top Contributors in jax-v0.2.28

mattjj
jakevdp
yashk2810
hawkinsp
froystig
Edenhofer
a-googler
tlu7
gnecula
GeraldCSC

Directory Browser for jax-v0.2.28

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

Release Notes Published

  • GitHub commits.
    • jax.jit(f).lower(...).compiler_ir() now defaults to the MHLO dialect if no dialect= is passed.
    • The jax.jit(f).lower(...).compiler_ir(dialect='mhlo') now returns an MLIR ir.Module object instead of its string representation.
GitClear uses cookies to ensure you get the best experience on our website. Learn more