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 numpy.testing.assert_allclose(),
which work directly with JAX arrays (#9620 ).
jax.test_util.JaxTestCase now sets jax_numpy_rank_promotion='raise' by default
(#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):
...
Fixed a performance problem if the indices passed to
{func}jax.numpy.take_along_axis were broadcasted ({jax-issue}[#10281](https://github.com/google/jax/issues/10281)).
{func}jax.scipy.special.expit and {func}jax.scipy.special.logit now
require their arguments to be scalars or JAX arrays. They also now promote
integer arguments to floating point.
The DeviceArray.tile() method is deprecated, because numpy arrays do not have a
tile() method. As a replacement for this, use {func}jax.numpy.tile
({jax-issue}[#10266](https://github.com/google/jax/issues/10266)).
jaxlib 0.3.7 (April 15, 2022)
Changes:
Linux wheels are now built conforming to the manylinux2014 standard, instead
of manylinux2010.
added {func}jax.random.loggamma & improved behavior of {func}jax.random.beta
and {func}jax.random.dirichlet for small parameter values ({jax-issue}[#9906](https://github.com/google/jax/issues/9906)).
the private lax_numpy submodule is no longer exposed in the jax.numpy namespace ({jax-issue}[#10029](https://github.com/google/jax/issues/10029)).
added array creation routines {func}jax.numpy.frombuffer, {func}jax.numpy.fromfunction,
and {func}jax.numpy.fromstring ({jax-issue}[#10049](https://github.com/google/jax/issues/10049)).
DeviceArray.copy() now returns a DeviceArray rather than a np.ndarray ({jax-issue}[#10069](https://github.com/google/jax/issues/10069))
added {func}jax.scipy.linalg.rsf2csf
Deprecations:
{func}jax.nn.normalize is being deprecated. Use {func}jax.nn.standardize instead ({jax-issue}[#9899](https://github.com/google/jax/issues/9899)).
{func}jax.tree_util.tree_multimap is deprecated. Use {func}jax.tree_util.tree_map instead ({jax-issue}[#5746](https://github.com/google/jax/issues/5746)).
jax.experimental.sharded_jit is deprecated. Use pjit instead.
jaxlib 0.3.5 (April 7, 2022)
Bug fixes
Fixed a bug where double-precision complex-to-real IRFFTs would mutate their
input buffers on GPU ({jax-issue}[#9946](https://github.com/google/jax/issues/9946)).
Fixed incorrect constant-folding of complex scatters ({jax-issue}[#10159](https://github.com/google/jax/issues/10159))
... (truncated)
Commits
1189c3c Merge pull request #10312 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 jax from 0.2.8 to 0.3.7.
Release notes
Sourced from jax's releases.
... (truncated)
Changelog
Sourced from jax's changelog.
... (truncated)
Commits
1189c3c
Merge pull request #10312 from hawkinsp:jaxlib6b33c55
2nd attempt at jax/jaxlib 0.3.7 release.a303e4b
Merge pull request #10266 from jakevdp:ndarray-tile6eec758
[JAX] Don't pass non-integer indices to jnp.take_along_axis7ffdac0
Skip remote_transfer_test on cloud TPU.a4b8a44
Merge pull request #10288 from YouJiacheng:patch-7be5c84d
Deprecate DeviceArray.tile method4ff6b1f
Fix PRNGKeyArray.broadcast_to with scalar shape470f58c
Merge pull request #10309 from hawkinsp:jaxlib52a97f2
Jax 0.3.7 and jaxlib 0.3.7 release.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)