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

Jax: jaxlib-v0.1.74 Release

Release date:
November 17, 2021
Previous version:
jaxlib-v0.1.73 (released October 18, 2021)
Magnitude:
14,226 Diff Delta
Contributors:
25 total committers
Data confidence:
Commits:
136 impacting commits

136 Commits in this Release

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

Adds an initial prototype of an alternate JAX compilation path that emits the MLIR MHLO/CHLO dialects instead of classic XLA HLO.
Authored November 11, 2021
[JAX] Port lax translation rules to updated XLA translation rule API.
Authored October 19, 2021
add internal `vmappable` interface (part 1)
Authored October 6, 2021
Move xla_bridge.constant to jax.interpreter.xla.pyval_to_ir_constant.
Authored October 19, 2021
bcoo_dot_general: support contraction along batch dimensions
Authored November 12, 2021
Add a prototype IREE backend for JAX.
Authored November 15, 2021
[JAX] move example libraries from `jax.experimental` into `jax.example_libraries`
Authored October 20, 2021
Adds support for batch dimensions for `lax.gather` when enable_xla=False. This allows us to, for instance, convert `jax.vmap(lax.dynamic_slice)`.
Authored November 2, 2021
[JAX] Added `jax.lax.linalg.qdwh`.
Authored October 29, 2021
[sparse] add todense() primitive for use in sparsify transform
Authored October 26, 2021
Initial GlobalShardedDeviceArray implementation.
Authored November 4, 2021
[sparse]: add tracer-based implementation of sparsify
Authored November 4, 2021
Delete xla_bridge.xla.dtype_to_etype, replace it with jax.interpreters.xla.dtype_to_primitive_type.
Authored October 19, 2021
Add gather/scatter mode support to jax2tf.
Authored November 2, 2021
[XLA:TPU] Adding approximate nearest neighbor search on TPU feature to JAX.
Authored October 19, 2021
Initial GSDA-pjit integration.
Authored November 12, 2021
[sparse] add autodiff & transpose rules for csr primitives
Authored November 2, 2021
make iter(DeviceArray) return DeviceArrays w/o sync
Authored July 22, 2020
Overload jnp ops which are polymorphic to an array's value and support PRNGKeys.
Authored October 25, 2021
[sparse] implement broadcasted sparse-dense multiplication
Authored October 21, 2021
Make a non-None platform mandatory during XLA translation.
Authored November 16, 2021
[jax2tf] Improve the TF constant sharing
Authored November 11, 2021
Internal change
Authored November 13, 2021
Output GSDAs from pjit if `jax_gsda_out` flag is enabled.
Authored November 13, 2021
[sparse] add sparse.random_bcoo() utility
Authored October 26, 2021
special.lpmn: use more canonical testing approach
Authored October 19, 2021
[sparse] register BCOO as a vmappable
Authored November 5, 2021
GSDA callback implementation for `from_batched_callback` and `from_batched_callback_with_devices`.
Authored November 5, 2021
add donated_invars to xla.XlaComputation
Authored November 16, 2021
add fft normalisation
Authored October 20, 2021
[sparse] add transpose rule for coo_matmat
Authored November 1, 2021
Document jax.lax.Precision
Authored November 8, 2021
jax.numpy: fix boolean indexing with Ellipsis
Authored November 1, 2021
Implements padding support for lax.reduce_window when enable_xla=False.
Authored October 20, 2021
[jax2tf] Improve the documentation for jax2tf and SavedModel
Authored November 4, 2021
Add jax.distributed.initialize for multi-host GPU.
Authored October 25, 2021
[sparse] avoid default jit-compilation of sparse array functions
Authored October 26, 2021
Add a default cuda installation path and more explicit installation paths for CUDA jaxlib.
Authored October 19, 2021
jnp.result_type: respect default types
Authored November 1, 2021
Implement lax.reduce() batching rule case where batch dimensions differ between operands.
Authored November 3, 2021

Browse Other Releases

Latest Pending Unreleased 😎
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 Ξ”
jaxlib-v0.1.73 Released October 18, 2021
0 Ξ”
jaxlib-v0.1.72 Released October 12, 2021
11,692 Ξ”
jaxlib-v0.1.71 Released September 1, 2021
0 Ξ”
jaxlib-v0.1.70 Released July 29, 2021
1,350 Ξ”
jaxlib-v0.1.69 Released July 9, 2021
6,728 Ξ”

Top Contributors in jaxlib-v0.1.74

hawkinsp
jakevdp
mattjj
a-googler
yashk2810
marcvanzee
froystig
tlu7
gnecula
LenaMartens

Directory Browser for jaxlib-v0.1.74

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