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

Jax: jaxlib-v0.4.2 Release

Release date:
January 25, 2023
Previous version:
jaxlib-v0.4.1 (released December 13, 2022)
Magnitude:
31,232 Diff Delta
Contributors:
44 total committers
Data confidence:
Commits:
297 impacting commits

297 Commits in this Release

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

Convert the notebook to a plain MD file
Authored December 15, 2022
(NFC) Prepare for migration from producing MHLO to producing StableHLO
Authored December 16, 2022
Delete the pjit 101 tutorial which is subsumed by https://jax.readthedocs.io/en/latest/notebooks/Distributed_arrays_and_automatic_parallelization.html
Authored January 21, 2023
Bump the minimum jaxlib version to 0.4.1
Authored December 19, 2022
[jax2tf] Adds tests for polymorphic input for Flax Module Harnesses
Authored January 10, 2023
Instructions to create custom operation for GPUs
Authored December 13, 2022
[jax2tf] Refactoring of shape_poly_test
Authored December 17, 2022
migrate internal dependencies from `jax.core` to `jax._src.core`
Authored December 17, 2022
Checkify: switch to initial-style
Authored December 6, 2022
[shape_poly] Generalize binary operations with symbolic dimensions
Authored January 18, 2023
[sparse] refactor/clean up test suite
Authored December 16, 2022
Remove experimental endpoints and update to point to 0.4.1
Authored December 13, 2022
[sparse] add nfold_vmap utility
Authored January 6, 2023
Delete `jax.experimental.callback`
Authored January 13, 2023
Add IO callback
Authored November 10, 2022
Multinodes CICD on GPUs using on-demand cluster and e2e tests using T5X
Authored September 26, 2022
Expose fp8 in jax dtypes and mlir builder
Authored January 14, 2023
Add `device` and `backend` API to `pjit` but resolve them away in `infer_params`
Authored December 14, 2022
[jax2tf] Improve handling of dimension polynomials used with JAX arrays
Authored December 28, 2022
Make `jit` a thin wrapper around `pjit` which ignores the mesh context manager (just like how it is today)
Authored January 13, 2023
Support `static_argnames` in pjit as a first step to merge `jit` and `pjit`
Authored December 13, 2022
[sparse] improve implementation of sparse.grad & sparse.value_and_grad
Authored December 28, 2022
[sparse] refactor tests to improve runtime
Authored January 20, 2023
[jax2tf] Ensure that dim_as_value returns int64 in x64 mode and int32 otherwise
Authored December 20, 2022
[sparse] Add bcsr dot_general
Authored December 19, 2022
Make the device assignment of outfeed configurable
Authored December 20, 2022
[sparse] add higher-level version of bcoo_extract & improve tests
Authored January 5, 2023
Fix scan and name_stack, rewrite cond to use jaxpr_to_checkify_jaxpr
Authored December 20, 2022
[jax2tf] Improves support for examples testing and adds three examples
Authored January 9, 2023
[sparse] bcoo_extract: add assume_unique keyword
Authored January 4, 2023
[sparse] handle general batch dimensions in bcoo transpose
Authored December 23, 2022
[jax2tf] Improves jax2tf (enable_xla=False) model testing logic
Authored January 5, 2023
[jax2tf] Improve handling of lax.pad and jnp.pad with polymorphic padding config
Authored December 29, 2022
Sharp Bits: add section on Dynamic shapes
Authored January 19, 2023
[sparse] Use SparseInfo for both bcoo and bcsr format
Authored December 16, 2022
[host_callback] Add device_index to hcb.call and add tests
Authored January 18, 2023
Add checkify.debug_check which is a noop outside of checkify
Authored December 12, 2022
Migrate pytestmark usage to new `@jtu.pytest_mark_if_available` decorator
Authored January 12, 2023
[jax2tf] Fixed the conversion of a function that contains an inner pjit
Authored January 8, 2023
[sparse] add CheckGradsSparse test utility
Authored December 15, 2022

Browse Other Releases

Latest Pending Unreleased 😎
jaxlib-v0.4.7 Released March 27, 2023
28,303 Ξ”
jaxlib-v0.4.6 Released March 9, 2023
21,059 Ξ”
jaxlib-v0.4.4 Released February 16, 2023
6,670 Ξ”
jaxlib-v0.4.3 Released February 8, 2023
14,421 Ξ”
jaxlib-v0.4.2 Released January 25, 2023
31,232 Ξ”
jaxlib-v0.4.1 Released December 13, 2022
592 Ξ”
jaxlib-v0.4.0 Released December 12, 2022
20,648 Ξ”
jaxlib-v0.3.25 Released November 15, 2022
8,314 Ξ”
jaxlib-v0.3.24 Released November 4, 2022
22,228 Ξ”
jaxlib-v0.3.22 Released October 11, 2022
22 Ξ”

Top Contributors in jaxlib-v0.4.2

yashk2810
jakevdp
gnecula
seanprime7
burmako
marcvanzee
hawkinsp
LenaMartens
sharadmv
froystig

Directory Browser for jaxlib-v0.4.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 about our cookie policy