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):
...
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))
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
... (truncated)
Commits
7f751c5 Update libtpu version for jax 0.3.5 release.
28cb44e Merge pull request #10184 from jakevdp:merge-bcoo-dot-general
96af4d5 Remove sharded_jit usage from jax2tf because sharded_jit is deprecated.
01e4fa8 [sparse] consolidate flavors of bcoo_dot_general
5522ed1 jax2tf: Support uint32 keys in rng_bit_generator.
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.5.
Release notes
Sourced from jax's releases.
... (truncated)
Changelog
Sourced from jax's changelog.
... (truncated)
Commits
7f751c5
Update libtpu version for jax 0.3.5 release.28cb44e
Merge pull request #10184 from jakevdp:merge-bcoo-dot-general96af4d5
Remove sharded_jit usage from jax2tf because sharded_jit is deprecated.01e4fa8
[sparse] consolidate flavors of bcoo_dot_general5522ed1
jax2tf: Support uint32 keys in rng_bit_generator.02fd875
Add init to PolyShape.b713d3c
Minor change to lax to support jax2tf shape polymorphic concatenation.cbdcdf7
[MHLO] Add MHLO lowerings for parallel collectives.2884215
Merge pull request #10167 from ROCmSoftwarePlatform:rocm_solver_api_consolida...8b3f039
Merge pull request #10039 from ajcr:add_scipy_linalg_rsf2csfDependabot 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)