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).
The functions jax.ops.index_update, jax.ops.index_add, which were
deprecated in 0.2.22, have been removed. Please use
the .at property on JAX arrays
instead, e.g., x.at[idx].set(y).
Moved jax.experimental.ann.approx_*_k into jax.lax. These functions are
optimized alternatives to jax.lax.top_k.
{func}jax.numpy.broadcast_arrays and {func}jax.numpy.broadcast_to now require scalar
or array-like inputs, and will fail if they are passed lists (part of {jax-issue}[#7737](https://github.com/google/jax/issues/7737)).
The standard jax[tpu] install can now be used with Cloud TPU v4 VMs.
pjit now works on CPU (in addition to previous TPU and GPU support).
jaxlib 0.3.2 (March 16, 2022)
Changes
XlaComputation.as_hlo_text() now supports printing large constants by
passing boolean flag print_large_constants=True.
Deprecations:
The .block_host_until_ready() method on JAX arrays has been deprecated.
Use .block_until_ready() instead.
jax.test_util.JaxTestCase and jax.test_util.JaxTestLoader are now deprecated.
The suggested replacement is to use parametrized.TestCase directly. For tests that
rely on custom asserts such as JaxTestCase.assertAllClose(), the suggested replacement
is to use standard numpy testing utilities such as {func}numpy.testing.assert_allclose(),
which work directly with JAX arrays ({jax-issue}[#9620](https://github.com/google/jax/issues/9620)).
jax.test_util.JaxTestCase now sets jax_numpy_rank_promotion='raise' by default
({jax-issue}[#9562](https://github.com/google/jax/issues/9562)). To recover the previous behavior, use the new
jax.test_util.with_config decorator:
@jtu.with_config(jax_numpy_rank_promotion='allow')
class MyTestCase(jtu.JaxTestCase):
...
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)
Bumps jaxlib from 0.1.61 to 0.3.2.
Release notes
Sourced from jaxlib's releases.
Changelog
Sourced from jaxlib's changelog.
... (truncated)
Commits
f47ce79
Merge pull request #9901 from jakevdp:fix-int-x6436dabf1
jnp.unique: avoid constructing arrays with explicit int645354a01
Merge pull request #9897 from jakevdp:numpy-doc846c480
Set in_positional_semantics should be GLOBAL for fully replicated values to a...4848c75
Adds use_auto_spmd_partitioning and propagates its value down the stack09105b2
Update outdated jax.numpy module docs057905b
Update WORKSPACE for jaxlib 0.3.2 release, take 31dd003e
Merge pull request #9890 from jakevdp:doc-eigh6ac05c0
Merge pull request #9889 from jakevdp:fix-segment-maxc66f5dd
DOC: add missing linalg functionality to docsDependabot 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)