Home » Oreps » Jax » All Releases » jax-v0.4.8 Release

google/jax

Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more

Watch

Basic Details Issues Addressed Top Contributors Directory Browser

Jax jax-v0.4.8

Jax: jax-v0.4.8 Release

Release date:

March 29, 2023

Previous version:

jax-v0.4.7 (released March 27, 2023)

Magnitude:

2,231 Diff Delta

Contributors:

11 total committers

Data confidence:

Commits:

26 Commits in this Release

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

Move jax.interpreters.partial_eval to jax._src.interpreters.partial_eval.

Authored March 27, 2023

Bump minimum jaxlib version from 0.4.6 to 0.4.7.

Authored March 28, 2023

[shape_poly] Refactor the computation of the dimension variables in native serialization

Authored March 12, 2023

Update the CUDA installation instructions.

Authored March 29, 2023

Remove global_arg_shapes from pmap since it was only used for sharded_jit and sharded_jit was removed from JAX a long time ago

Authored March 29, 2023

Deprecate FROM_GDA and remove its support from pjit's code since jax.Array inside pjit has sharding inference capabilities by default.

Authored March 28, 2023

Add deprecation warnings for several top-level jax imports

Authored March 28, 2023

Remove Cuda 11.4 support. JAX from 0.4.8 release will support cuda 11.8 and cuda 12

Authored March 28, 2023

Turn on PJRT C API by default.

Authored March 28, 2023

Split core.py and several files in an SCC with it into a separate Bazel build target.

Authored March 29, 2023

Add jaxlib version guard for rnn test

Authored March 27, 2023

Set coordinator address to allow it to later be used to initialize OCDBT coordinator server. Allow user to pass ts.Context when serializing or deserializing.

Authored March 28, 2023

Remove MeshPspecSharding since it has been more than 3 months since it was deprecated (Nov 2, 2022). The replacement name is NamedSharding.

Authored March 28, 2023

jax.typing: recommend instance check in Python 3.10 or newer

Authored March 27, 2023

Finish jax and jaxlib 0.4.7 release

Authored March 27, 2023

Remove jax2tf experimental_native_lowering.

Authored March 28, 2023

remove scale in wald docstring

Authored March 29, 2023

[sparse] fix coo efficiency warning

Authored March 27, 2023

Fixed spelling error in msgs

Authored March 28, 2023

CI: add numpy & scipy to mypy env

Authored March 27, 2023

Recommend --local_test_jobs in bazel test command line on GPU.

Authored March 29, 2023

Jax 0.4.7 has been released so assert that length of warnings is 1 in test_cache_read_warning

Authored March 28, 2023

Add version constraints to CUDA pip wheel dependencies.

Authored March 29, 2023

Silence some pytype errors.

Authored March 28, 2023

Add minimal pyproject.toml specifying build system

Authored March 29, 2023

Internal Code Change

Authored March 29, 2023

Browse Other Releases

Latest Pending Unreleased 😎

jax-v0.4.11-rc Released May 31, 2023

6,799 Δ

jax-v0.4.10 Released May 11, 2023

0 Δ

jax-v0.4.10-rc Released May 11, 2023

1,233 Δ

jax-v0.4.9 Released May 9, 2023

1,058 Δ

jax-v0.4.9-rc Released May 9, 2023

23,579 Δ

jax-v0.4.8 Released March 29, 2023

2,231 Δ

jax-v0.4.7 Released March 27, 2023

0 Δ

jax-v0.4.7-rc1 Released March 27, 2023

0 Δ

jax-v0.4.7-rc Released March 27, 2023

19,399 Δ

jax-v0.4.6 Released March 9, 2023

0 Δ

Top Contributors in jax-v0.4.8

hawkinsp

skye

yashk2810

gnecula

jakevdp

sharadmv

cpgaffney1

JiaYaobo

mehdiataei

rchen152

Directory Browser for jax-v0.4.8

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