Out-of-bounds indices to jax.ops.segment_sum will now be handled with
FILL_OR_DROP semantics, as documented. This primarily afects the
reverse-mode derivative, where gradients corresponding to out-of-bounds
indices will now be returned as 0. (#8634).
jax2tf will force the converted code to use XLA for the code fragments
under jax.jit, e.g., most jax.numpy functions ({jax-issue}[#7839](https://github.com/google/jax/issues/7839)).
jaxlib 0.1.74 (Nov 17, 2021)
Enabled peer-to-peer copies between GPUs. Previously, GPU copies were bounced via
the host, which is usually slower.
Added experimental MLIR Python bindings for use by JAX.
jax.random.choice and jax.random.permutation now support
multidimensional arrays and an optional axis argument ({jax-issue}[#8158](https://github.com/google/jax/issues/8158))
Breaking changes:
jax.numpy.take and jax.numpy.take_along_axis now require array-like inputs
(see {jax-issue}[#7737](https://github.com/google/jax/issues/7737))
jaxlib 0.1.73 (Oct 18, 2021)
Multiple cuDNN versions are now supported for jaxlib GPU cuda11 wheels.
cuDNN 8.2 or newer. We recommend using the cuDNN 8.2 wheel if your cuDNN
installation is new enough, since it supports additional functionality.
cuDNN 8.0.5 or newer.
... (truncated)
Commits
2f4bd11 Merge pull request #8842 from google:yashk2810-patch-3
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.1.75.
Release notes
Sourced from jaxlib's releases.
Changelog
Sourced from jaxlib's changelog.
... (truncated)
Commits
2f4bd11
Merge pull request #8842 from google:yashk2810-patch-3b0b853c
Update WORKSPACE06cd1fe
Move dtype canonicalization out of core.AbstractValue subclasses.56f029f
[jax] Add computation name to cache hit logging.803d3f2
Merge pull request #8829 from LenaMartens:changelist/414499911bbf1a9b
Add checkify rule for scan.068bfbf
Remove the TODO0bb7d20
Move serialization/de-serialization of GDA into jax.5de2cec
Threadis_gda
to lower_mesh_computation. This is required because to lift t...01dd34b
[MLIR] Use XLA implementation of sinh, instead of the CHLO lowering.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)