Extend the :mod:jax.experimental.loops module with support for pytrees. Improved
error checking and error messages.
Add :func:jax.experimental.enable_x64 and :func:jax.experimental.disable_x64.
These are context managers which allow X64 mode to be temporarily enabled/disabled
within a session.
Breaking changes:
:func:jax.ops.segment_sum now drops segment IDs that are out of range rather
than wrapping them into the segment ID space. This was done for performance
reasons.
Commits
7495f71 Merge pull request #5614 from hawkinsp:jaxlib
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.59 to 0.1.60.
Changelog
Sourced from jaxlib's changelog.
Commits
7495f71
Merge pull request #5614 from hawkinsp:jaxlib3829bb9
Release jaxlib v0.1.60.3fdc14a
Merge pull request #5578 from gnecula:tf_sharded_jit5616916
Merge pull request #5602 from apaszke:xmap-explicit-sizes9c72582
Merge pull request #5584 from jakevdp:fix-multivariate-normal024ac7c
Merge pull request #5583 from apaszke:logsumexp-namede27f1e9
Merge pull request #5600 from apaszke:named-one-hota8b1f5f
Add axis_sizes to xmapf812402
Add support for named axes in jax.nn.one_hot78c54ae
Merge pull request #5589 from ahoenselaar:changelist/355051176Dependabot 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)