{func}jax.scipy.stats.chi2 is now available as a distribution with logpdf and pdf methods.
{func}jax.scipy.stats.betabinom is now available as a distribution with logpmf and pmf methods.
Added {func}jax.experimental.jax2tf.call_tf to call TensorFlow functions
from JAX ({jax-issue}[#5627](https://github.com/google/jax/issues/5627))
and README).
Extended the batching rule for lax.pad to support batching of the padding values.
Bug fixes:
{func}jax.numpy.take properly handles negative indices ({jax-issue}[#5768](https://github.com/google/jax/issues/5768))
Breaking changes:
JAX's promotion rules were adjusted to make promotion more consistent and
invariant to JIT. In particular, binary operations can now result in weakly-typed
values when appropriate. The main user-visible effect of the change is that
some operations result in outputs of different precision than before; for
example the expression jnp.bfloat16(1) + 0.1 * jnp.arange(10)
previously returned a float64 array, and now returns a bfloat16 array.
JAX's type promotion behavior is described at {ref}type-promotion.
{func}jax.numpy.linspace now computes the floor of integer values, i.e.,
rounding towards -inf rather than 0. This change was made to match NumPy
1.20.0.
{func}jax.numpy.i0 no longer accepts complex numbers. Previously the
function computed the absolute value of complex arguments. This change was
made to match the semantics of NumPy 1.20.0.
Several {mod}jax.numpy functions no longer accept tuples or lists in place
of array arguments: {func}jax.numpy.pad, :funcjax.numpy.ravel,
{func}jax.numpy.repeat, {func}jax.numpy.reshape.
In general, {mod}jax.numpy functions should be used with scalars or array arguments.
jaxlib 0.1.62 (March 9 2021)
New features:
jaxlib wheels are now built to require AVX instructions on x86-64 machines
by default. If you want to use JAX on a machine that doesn't support AVX,
you can build a jaxlib from source using the --target_cpu_features flag
to build.py. --target_cpu_features also replaces
--enable_march_native.
Commits
2947f56 Merge pull request #6120 from skye:workspace
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)
dependabot[bot]
commented
3 years ago
Bumps jaxlib from 0.1.61 to 0.1.64.
Changelog
Sourced from jaxlib's changelog.
Commits
2947f56Merge pull request #6120 from skye:workspace6965ab2Update WORKSPACE for profiler fix6ac9463Merge pull request #6106 from jakevdp:lax-test6041e1blax_test: increase test coverage0198f64Merge pull request #6103 from jakevdp:lax-const1851d7cMerge pull request #6113 from gnecula:jax2tf_serving_2255b1e3Merge pull request #6116 from gnecula:hcb_buffer_size441178bFix flake52408b3Move the flag definitions to config8f846a3[host_callback] Enable configuring the outfeed receiver buffer sizeDependabot 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)