search

asmith26 / jax_toolkit

A collection of jax functions to help with common machine/deep learning related functionality.
https://asmith26.github.io/jax_toolkit/
Apache License 2.0
5 stars 0 forks source link

Bump jax from 0.2.5 to 0.2.6 #112

Closed dependabot[bot] closed 3 years ago

dependabot[bot] commented 3 years ago

Bumps jax from 0.2.5 to 0.2.6.

Changelog

Sourced from jax's changelog.

jax 0.2.6 (Nov 18 2020)

  • GitHub commits <https://github.com/google/jax/compare/jax-v0.2.5...jax-v0.2.6>_.

  • New Features:

    • Add support for shape-polymorphic tracing for the jax.experimental.jax2tf converter. See README.md <https://github.com/google/jax/blob/master/jax/experimental/jax2tf/README.md>_.

  • Breaking change cleanup

    • Raise an error on non-hashable static arguments for jax.jit and xla_computation. See cb48f42 <https://github.com/google/jax/commit/cb48f42>_.

    • Improve consistency of type promotion behavior ([#4744](https://github.com/google/jax/issues/4744) <https://github.com/google/jax/pull/4744>_):

      • Adding a complex Python scalar to a JAX floating point number respects the precision of the JAX float. For example, jnp.float32(1) + 1j now returns complex64, where previously it returned complex128.
      • Results of type promotion with 3 or more terms involving uint64, a signed int, and a third type are now independent of the order of arguments. For example: jnp.result_type(jnp.uint64, jnp.int64, jnp.float16) and jnp.result_type(jnp.float16, jnp.uint64, jnp.int64) both return float16, where previously the first returned float64 and the second returned float16.

    • The contents of the (undocumented) jax.lax_linalg linear algebra module are now exposed publicly as jax.lax.linalg.

    • jax.random.PRNGKey now produces the same results in and out of JIT compilation ([#4877](https://github.com/google/jax/issues/4877) <https://github.com/google/jax/pull/4877>_). This required changing the result for a given seed in a few particular cases:

      • With jax_enable_x64=False, negative seeds passed as Python integers now return a different result outside JIT mode. For example, jax.random.PRNGKey(-1) previously returned [4294967295, 4294967295], and now returns [0, 4294967295]. This matches the behavior in JIT.
      • Seeds outside the range representable by int64 outside JIT now result in an OverflowError rather than a TypeError. This matches the behavior in JIT.

      To recover the keys returned previously for negative integers with jax_enable_x64=False outside JIT, you can use::

      key = random.PRNGKey(-1).at[0].set(0xFFFFFFFF)

    • DeviceArray now raises RuntimeError instead of ValueError when trying to access its value while it has been deleted.

jaxlib 0.1.58 (Unreleased)

  • Fixed a bug that meant JAX sometimes return platform-specific types (e.g.,

... (truncated)

Commits
  • 0978ccb Merge pull request #4946 from jakevdp:jax-026
  • 050b879 Merge pull request #4926 from bchetioui:test_transpose
  • ef4adb5 update version & CHANGELOG for jax v0.2.6
  • 5afc775 Merge pull request #4868 from sharadmv:variadic_reduce
  • 3c671e8 Merge pull request #4940 from jakevdp:var-warning
  • 8cb62f2 Merge pull request #4878 from gnecula:revisit_poly
  • 3ba7f42 Merge pull request #4944 from hawkinsp:sourceinfo
  • 13db2c3 Fix mypy error caused by cyclic import dependency.
  • f939e71 Merge pull request #4943 from hawkinsp:sourceinfo
  • e38f96d Use the jax module to determine the jax source directory base, rather than th...
  • Additional commits viewable in compare view


Dependabot compatibility score

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

Looks like jax is up-to-date now, so this is no longer needed.