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

Jax: jax-v0.3.11 Release

Release date:
May 14, 2022
Previous version:
jax-v0.3.10 (released May 4, 2022)
Magnitude:
9,164 Diff Delta
Contributors:
25 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.

Merge remaining CUDA and ROCM Python code.
Authored May 6, 2022
Add sequencing effects design note
Authored May 10, 2022
Enable print in while loops/scan
Authored May 4, 2022
add scan dce rule tests, fix bugs
Authored May 1, 2022
improve partial_eval_jaxpr_custom
Authored May 7, 2022
[JAX] Split the "gpu" platform in internal JAX usage into separate "cuda" and "rocm" platforms.
Authored May 5, 2022
[sparse] add bcoo_update_layout utility
Authored May 5, 2022
add core.closed_call_p
Authored May 14, 2022
Rollback of PR #10393 "Improve performance of array integer indexing"
Authored May 10, 2022
[MHLO] Migrate GPU select_and_gather_add translation rule to MHLO.
Authored May 9, 2022
Change non-array arguments to jax.lax.linalg functions to be keyword-only arguments.
Authored May 11, 2022
Disabled one and enabled several unit tests for ROCm.
Authored May 10, 2022
DOC: update myst-nb to v0.15.0
Authored May 9, 2022
Attach keepalive to pmap executable
Authored May 10, 2022
Add support for padded arrays in QDWH algorithm.
Authored May 13, 2022
Add a sort_eigenvalues option to lax.linalg.eigh().
Authored May 11, 2022
Enable effects for cond
Authored May 4, 2022
Enable effects for cond in while loop
Authored May 4, 2022
Add jax.scipy.linalg.funm
Authored April 26, 2022
[linalg] Add svd on zero matrix.
Authored May 9, 2022
Enable batching rule for debug_print
Authored May 5, 2022
jax.scipy.sparse.linalg: support sparse matrices as operators
Authored May 5, 2022
Consolidate hip_prng and cuda_prng.
Authored May 5, 2022
Speedup `_expand_bool_indices` when passing basic integer indices
Authored April 20, 2022
Internal change
Authored May 9, 2022
make core_test.py pass with core.call
Authored May 5, 2022
remove count attribute and total_ordering from core.Var
Authored May 4, 2022
[linalg] Update qdwh to prevent underflow in norm estimation.
Authored May 6, 2022
Speedup index computations for integer indexing
Authored April 12, 2022
[jax2tf] Update JAX limitations.
Authored May 9, 2022
minor edits to clarify pjit docs
Authored May 8, 2022
fix ad_checkpoint.checkpoint vmap rule
Authored May 5, 2022
Avoid generating trivial gathers when reversing array
Authored May 9, 2022
Simplify `_normalize_index`
Authored April 21, 2022
Update custom interpreter tutorial
Authored May 11, 2022
fix grad(..., argnums=-1), regressed in #10453
Authored May 11, 2022
Do not call `concrete_aval` for basic integer index checks
Authored April 20, 2022
Enable AD rules for `debug_print`
Authored May 5, 2022
tweak mlir shape_tensor helper, fewer MHLO ops
Authored May 12, 2022
[sparse] incremental improvement to docs
Authored May 4, 2022

Browse Other Releases

Latest Pending Unreleased 😎
jax-v0.3.15-rc Released July 22, 2022
18,821 Ξ”
jax-v0.3.14 Released June 27, 2022
25,706 Ξ”
jax-v0.3.13 Released May 16, 2022
190 Ξ”
jax-v0.3.12 Released May 16, 2022
13 Ξ”
jax-v0.3.11 Released May 14, 2022
9,164 Ξ”
jax-v0.3.10 Released May 4, 2022
1,645 Ξ”
jax-v0.3.9 Released May 3, 2022
878 Ξ”
jax-v0.3.8 Released April 30, 2022
11,151 Ξ”
jax-v0.3.7 Released April 15, 2022
4,073 Ξ”
jax-v0.3.6 Released April 13, 2022
5,775 Ξ”

Top Contributors in jax-v0.3.11

hawkinsp
sharadmv
mattjj
jakevdp
lgeiger
tlu7
levskaya
rsanthanam-amd
ajcr
gnecula

Directory Browser for jax-v0.3.11

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

Release Notes Published

  • Changes
    • {func}jax.lax.eigh now accepts an optional sort_eigenvalues argument that allows users to opt out of eigenvalue sorting on TPU.
  • Deprecations
    • Non-array arguments to functions in {mod}jax.lax.linalg are now marked keyword-only. As a backward-compatibility step passing keyword-only arguments positionally yields a warning, but in a future JAX release passing keyword-only arguments positionally will fail. However, most users should prefer to use {mod}jax.numpy.linalg instead.
    • {func}jax.scipy.linalg.polar_unitary, which was a JAX extension to the scipy API, is deprecated. Use {func}jax.scipy.linalg.polar instead.
GitClear uses cookies to ensure you get the best experience on our website. Learn more