Jax: jaxlib-v0.4.1 Release

Release date:
December 13, 2022
Previous version:
jaxlib-v0.4.0 (released December 12, 2022)
Magnitude:
592 Diff Delta
Contributors:
4 total committers
Data confidence:
Commits:

Top Contributors in jaxlib-v0.4.1

gnecula
jakevdp
yashk2810
hawkinsp

Directory Browser for jaxlib-v0.4.1

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

Release Notes Published

  • Changes
    • Support for Python 3.7 has been dropped, in accordance with JAX's {ref}version-support-policy.
    • The behavior of XLA_PYTHON_CLIENT_MEM_FRACTION=.XX has been changed to allocate XX% of the total GPU memory instead of the previous behavior of using currently available GPU memory to calculate preallocation. Please refer to GPU memory allocation for more details.
    • The deprecated method .block_host_until_ready() has been removed. Use .block_until_ready() instead.