Ensure that check_jaxpr does not perform FLOPS. See [#4650](https://github.com/google/jax/issues/4650) <https://github.com/google/jax/pull/4650>_.
Expanded the set of JAX primitives converted by jax2tf.
See primitives_with_limited_support.md <https://github.com/google/jax/blob/master/jax/experimental/jax2tf/primitives_with_limited_support.md>_.
Add support for remat to jax.experimental.host_callback. See [#4608](https://github.com/google/jax/issues/4608) <https://github.com/google/jax/pull/4608>_.
Commits
a3a55a9 Merge pull request #4847 from hawkinsp:py39
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.56 to 0.1.57.
Changelog
Sourced from jaxlib's changelog.
Commits
a3a55a9
Merge pull request #4847 from hawkinsp:py392c6f932
Add Python 3.9 support to jaxlib build.307b528
Merge pull request #4814 from apaszke:sharding-spec8c1fbdc
Make ShardingSpec more flexible08131d4
Merge pull request #4844 from hawkinsp:manylinuxdadf732
Fix manylinux2010 compliance of GPU wheels.78e7095
Merge pull request #4846 from zhangqiaorjc:profiler_fix_note5e85095
Add a CHANGELOG note about profiler bug fix.ca766ca
Merge pull request #4822 from ayush-1506:4775dcc81c3
Merge pull request #4845 from hawkinsp:xlaDependabot 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)