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

Jax: jaxlib-v0.3.2 Release

Release date:
March 15, 2022
Previous version:
jaxlib-v0.3.0 (released February 10, 2022)
Magnitude:
19,405 Diff Delta
Contributors:
33 total committers
Data confidence:
Commits:
202 impacting commits

202 Commits in this Release

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

Update JAX to use new math libraries in ROCm-5.0.
Authored February 15, 2022
Add separate mechanism for threading name stacks to the lowering
Authored October 28, 2021
Add some functions for spectral analysis.
Authored February 3, 2022
JaxTestCase: set numpy_rank_promotion='raise' by default
Authored February 1, 2022
Copybara import of the project:
Authored February 11, 2022
[sparse] change call signature of coo primitive wrappers
Authored March 7, 2022
re-implement `custom_transpose` without upfront staging.
Authored January 21, 2022
Xmap GDA integration. Non-contiguous mesh is allowed!
Authored February 17, 2022
Increase minimum jaxlib version to 0.3.0.
Authored March 4, 2022
Document jax.nn.initializers.
Authored March 7, 2022
Rollback of:
Authored March 2, 2022
remove `_const` from public `jax.lax` module
Authored March 7, 2022
[sparse] track sorted columns for COO GPU lowerings
Authored March 9, 2022
Set in_positional_semantics should be GLOBAL for fully replicated values to avoid recompilation.
Authored March 15, 2022
Try adding support for nesting with_sharding_constraint in MANUAL xmaps
Authored February 15, 2022
[remove units] make JaxprTrace.process_call not introduce units
Authored February 7, 2022
lax_numpy.py: factor ufuncs into their own private submodule
Authored March 15, 2022
DOC: add ability to document extra_params within _wraps
Authored February 10, 2022
Factor-out pieces of lax_numpy.py
Authored March 1, 2022
JaxTestCase: set jax_numpy_rank_promotion='raise' by default
Authored February 14, 2022
Make `mesh_axes` on GDA strict by only allowing PartitionSpecs to be consistent with pjit.
Authored March 7, 2022
remove `_convert_element_type` from public `jax.lax` module
Authored March 10, 2022
Merge `mesh` and `Mesh`. Make `Mesh` a context manager + class so that it can be used in the following ways:
Authored February 17, 2022
Always treat all mesh axes controlled by xmap as MANUAL
Authored February 22, 2022
Refactor Jaxpr pretty-printing to use a `JaxprPpSettings` named tuple
Authored March 9, 2022
Implement the JAX transfer guard API
Authored February 14, 2022
Remove jax.ops.index... functions.
Authored February 17, 2022
Add a partial_eval_jaxpr_custom_rule for xmap
Authored February 15, 2022
Add the input avals to Lowered and Compiled.
Authored March 9, 2022
[sparse]: COO: check for sorted rows before cusparse lowering
Authored March 8, 2022
[JAX] Update ann to use XLA based fallback ApproxTopK.
Authored March 15, 2022
Documentation improvements.
Authored March 8, 2022
Add an API compatibility policy.
Authored February 17, 2022
lax_numpy.py: factor out some common utilities
Authored March 10, 2022
Improve TPU v2 and v3 mesh_utils.create_device_mesh logic.
Authored March 8, 2022
* Make _old_env thread local so that it can be used in multiple threads.
Authored February 18, 2022
cleanup: remove _constant_like in favor of lax._const
Authored March 2, 2022
convert to doctest
Authored March 3, 2022
implement `jnp.expand_dims` and `jnp.stack` for PRNGKeyArrays
Authored February 17, 2022
Turn execute_replicated into a class so we can access its fields.
Authored February 18, 2022

Browse Other Releases

Latest Pending Unreleased 😎
jaxlib-v0.3.15 Released July 22, 2022
16,976 Ξ”
jaxlib-v0.3.14 Released June 27, 2022
33,973 Ξ”
jaxlib-v0.3.10 Released May 4, 2022
13,336 Ξ”
jaxlib-v0.3.7 Released April 15, 2022
10,780 Ξ”
jaxlib-v0.3.5 Released April 7, 2022
9,400 Ξ”
jaxlib-v0.3.2 Released March 15, 2022
19,405 Ξ”
jaxlib-v0.3.0 Released February 10, 2022
8,876 Ξ”
jaxlib-v0.1.76 Released January 27, 2022
15,529 Ξ”
jaxlib-v0.1.75 Released December 7, 2021
17,697 Ξ”
jaxlib-v0.1.74 Released November 17, 2021
14,226 Ξ”

Top Contributors in jaxlib-v0.3.2

jakevdp
yashk2810
reza-rahimi-144f
hawkinsp
froystig
a-googler
sharadmv
apaszke
yotarok
mattjj

Directory Browser for jaxlib-v0.3.2

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