Linux wheels are now built conforming to the manylinux2014 standard, instead of manylinux2010.
Jaxlib release v0.3.0
Changes
Bazel 5.0.0 is now required to build jaxlib.
jaxlib version has been bumped to 0.3.0. Please see the design doc
for the explanation.
Jaxlib v0.1.76
New features
Includes precompiled SASS for NVidia compute capability 8.0 GPUS
(e.g. A100). Removes precompiled SASS for compute capability 6.1 so as not
to increase the number of compute capabilities: GPUs with compute capability
6.1 can use the 6.0 SASS.
With jaxlib 0.1.76, JAX uses the MHLO MLIR dialect as its primary target compiler IR
by default.
Breaking changes
Support for NumPy 1.18 has been dropped, per the deprecation policy.
Please upgrade to a supported NumPy version.
Bug fixes
Fixed a bug where apparently identical pytreedef objects constructed by different routes
do not compare as equal (#9066).
The JAX jit cache requires two static arguments to have identical types for a cache hit (#9311).
JaxTestCase and JaxTestLoader have been removed from jax.test_util. These
classes have been deprecated since v0.3.1 ({jax-issue}[#11248](https://github.com/google/jax/issues/11248)).
Binary operations between JAX arrays and built-in collections (dict, list, set, tuple)
now raise a TypeError in all cases. Previously some cases (particularly equality and inequality)
would return boolean scalars inconsistent with similar operations in NumPy ({jax-issue}[#11234](https://github.com/google/jax/issues/11234)).
{mod}jax.tree_util routines accessed as top-level JAX package imports are now deprecated, and
will be removed in a future JAX release in accordance with the {ref}api-compatibility policy:
{func}jax.treedef_is_leaf is deprecated in favor of {func}jax.tree_util.treedef_is_leaf
{func}jax.tree_flatten is deprecated in favor of {func}jax.tree_util.tree_flatten
{func}jax.tree_leaves is deprecated in favor of {func}jax.tree_util.tree_leaves
{func}jax.tree_map is deprecated in favor of {func}jax.tree_util.tree_map
{func}jax.tree_structure is deprecated in favor of {func}jax.tree_util.tree_structure
{func}jax.tree_transpose is deprecated in favor of {func}jax.tree_util.tree_transpose
{func}jax.tree_unflatten is deprecated in favor of {func}jax.tree_util.tree_unflatten
The sym_pos argument of {func}jax.scipy.linalg.solve is deprecated in favor of assume_a='pos',
following a similar deprecation in {func}scipy.linalg.solve.
Deprecations:
{func}jax.mask {func}jax.shapecheck APIs have been removed.
See {jax-issue}[#11557](https://github.com/google/jax/issues/11557).
{func}jax.experimental.compilation_cache.initialize_cache does not support
max_cache_size_ bytes anymore and will not get that as an input.
JAX_PLATFORMS now raises an exception when platform initialization fails.
Changes
Fixed compatibility problems with NumPy 1.23.
{func}jax.numpy.linalg.slogdet now accepts an optional method argument
that allows selection between an LU-decomposition based implementation and
an implementation based on QR decomposition.
{func}jax.numpy.linalg.qr now supports mode="raw".
pickle, copy.copy, and copy.deepcopy now have more complete support when
used on jax arrays ({jax-issue}[#10659](https://github.com/google/jax/issues/10659)). In particular:
pickle and deepcopy previously returned np.ndarray objects when used
on a DeviceArray; now DeviceArray objects are returned. For deepcopy,
the copied array is on the same device as the original. For pickle the
deserialized array will be on the default device.
Within function transformations (i.e. traced code), deepcopy and copy
previously were no-ops. Now they use the same mechanism as DeviceArray.copy().
Calling pickle on a traced array now results in an explicit
ConcretizationTypeError.
The implementation of singular value decomposition (SVD) and
symmetric/Hermitian eigendecomposition should be significantly faster on
... (truncated)
Commits
1a7c883 Merge pull request #11589 from skye:workspace
26fbeb6 Update WORKSPACE and libtpu version for jaxlib 0.3.15, take 3
e121e81 Merge pull request #11536 from sharadmv:colab-debugger
0b6657e Merge pull request #11556 from RuffaloLavoisier:tYpO
4870710 Enable debugging callbacks with pjit on TPU
8a67734 Merge pull request #11579 from sharadmv:fix-effects
7f0b917 Merge pull request #11575 from gnecula:ds_progress
24134ec Merge pull request #11425 from pschuh:pjit-bugfix
540ee56 Merge pull request #11576 from jakevdp:searchsorted-alt
d6c172d Fix PE not allowing double JIT-ted effectful functions
Dependabot will resolve any conflicts with this PR as long as you don't alter it yourself. You can also trigger a rebase manually by commenting @dependabot rebase.
Dependabot commands and options
You can trigger Dependabot actions by commenting on this PR:
- `@dependabot rebase` will rebase this PR
- `@dependabot recreate` will recreate this PR, overwriting any edits that have been made to it
- `@dependabot merge` will merge this PR after your CI passes on it
- `@dependabot squash and merge` will squash and merge this PR after your CI passes on it
- `@dependabot cancel merge` will cancel a previously requested merge and block automerging
- `@dependabot reopen` will reopen this PR if it is closed
- `@dependabot close` will close this PR and stop Dependabot recreating it. You can achieve the same result by closing it manually
- `@dependabot ignore this major version` will close this PR and stop Dependabot creating any more for this major version (unless you reopen the PR or upgrade to it yourself)
- `@dependabot ignore this minor version` will close this PR and stop Dependabot creating any more for this minor version (unless you reopen the PR or upgrade to it yourself)
- `@dependabot ignore this dependency` will close this PR and stop Dependabot creating any more for this dependency (unless you reopen the PR or upgrade to it yourself)
dependabot[bot]
commented
2 years ago
Bumps jaxlib from 0.1.61 to 0.3.15.
Release notes
Sourced from jaxlib's releases.
... (truncated)
Changelog
Sourced from jaxlib's changelog.
... (truncated)
Commits
1a7c883Merge pull request #11589 from skye:workspace26fbeb6Update WORKSPACE and libtpu version for jaxlib 0.3.15, take 3e121e81Merge pull request #11536 from sharadmv:colab-debugger0b6657eMerge pull request #11556 from RuffaloLavoisier:tYpO4870710Enable debugging callbacks with pjit on TPU8a67734Merge pull request #11579 from sharadmv:fix-effects7f0b917Merge pull request #11575 from gnecula:ds_progress24134ecMerge pull request #11425 from pschuh:pjit-bugfix540ee56Merge pull request #11576 from jakevdp:searchsorted-altd6c172dFix PE not allowing double JIT-ted effectful functionsDependabot will resolve any conflicts with this PR as long as you don't alter it yourself. You can also trigger a rebase manually by commenting
@dependabot rebase.Dependabot commands and options
You can trigger Dependabot actions by commenting on this PR: - `@dependabot rebase` will rebase this PR - `@dependabot recreate` will recreate this PR, overwriting any edits that have been made to it - `@dependabot merge` will merge this PR after your CI passes on it - `@dependabot squash and merge` will squash and merge this PR after your CI passes on it - `@dependabot cancel merge` will cancel a previously requested merge and block automerging - `@dependabot reopen` will reopen this PR if it is closed - `@dependabot close` will close this PR and stop Dependabot recreating it. You can achieve the same result by closing it manually - `@dependabot ignore this major version` will close this PR and stop Dependabot creating any more for this major version (unless you reopen the PR or upgrade to it yourself) - `@dependabot ignore this minor version` will close this PR and stop Dependabot creating any more for this minor version (unless you reopen the PR or upgrade to it yourself) - `@dependabot ignore this dependency` will close this PR and stop Dependabot creating any more for this dependency (unless you reopen the PR or upgrade to it yourself)