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.8 to 0.3.1 #268

Closed dependabot[bot] closed 2 years ago

dependabot[bot] commented 2 years ago

Bumps jax from 0.2.8 to 0.3.1.

Release notes

Sourced from jax's releases.

Jax release v0.3.1

  • Changes:
    • 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 jax.scipy.linalg.schur, jax.scipy.linalg.sqrtm, jax.scipy.signal.csd, jax.scipy.signal.stft, jax.scipy.signal.welch.

Jax release v0.3.0

  • Changes
    • jax version has been bumped to 0.3.0. Please see the design doc for the explanation.

Jaxlib release v0.3.0

  • Changes
    • Bazel 5.0.0 is now required to build jaxlib.
    • jaxlib version has been bumped to 0.3.0. Please see the design doc for the explanation.

JAX release v0.2.28

  • GitHub commits.
    • jax.jit(f).lower(...).compiler_ir() now defaults to the MHLO dialect if no dialect= is passed.
    • The jax.jit(f).lower(...).compiler_ir(dialect='mhlo') now returns an MLIR ir.Module object instead of its string representation.

JAX release v0.2.27

  • GitHub commits.

  • Breaking changes:

    • Support for NumPy 1.18 has been dropped, per the [deprecation policy](https://jax.readthedocs.io/en/latest/ deprecation.html). Please upgrade to a supported NumPy version.
    • The host_callback primitives have been simplified to drop the special autodiff handling for hcb.id_tap and id_print. From now on, only the primals are tapped. The old behavior can be obtained (for a limited time) by setting the JAX_HOST_CALLBACK_AD_TRANSFORMS environment variable, or the --flax_host_callback_ad_transforms flag. Additionally, added documentation for how to implement the old behavior using JAX custom AD APIs ({jax-issue}[#8678](https://github.com/google/jax/issues/8678)).
    • Sorting now matches the behavior of NumPy for 0.0 and NaN regardless of the bit representation. In particular, 0.0 and -0.0 are now treated as equivalent, where previously -0.0 was treated as less than 0.0. Additionally all NaN representations are now treated as equivalent and sorted to the end of the array. Previously negative NaN values were sorted to the front of the array, and NaN values with different internal bit representations were not treated as equivalent, and were sorted according to those bit patterns ({jax- issue}[#9178](https://github.com/google/jax/issues/9178)).
    • {func}jax.numpy.unique now treats NaN values in the same way as np.unique in NumPy versions 1.21 and newer: at most one NaN value will appear in the uniquified output ({jax-issue}9184).

  • Bug fixes:

    • host_callback now supports ad_checkpoint.checkpoint ({jax-issue}[#8907](https://github.com/google/jax/issues/8907)).

  • New features:

... (truncated)

Changelog

Sourced from jax's changelog.

jaxlib 0.3.1 (Unreleased)

  • Changes

jax 0.3.1 (Feb 18, 2022)

  • GitHub commits.

  • Changes:

    • 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 {func}numpy.testing.assert_allclose(), which work directly with JAX arrays ({jax-issue}[#9620](https://github.com/google/jax/issues/9620)).
    • jax.test_util.JaxTestCase now sets jax_numpy_rank_promotion='raise' by default ({jax-issue}[#9562](https://github.com/google/jax/issues/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.scipy.linalg.schur, {func}jax.scipy.linalg.sqrtm, {func}jax.scipy.signal.csd, {func}jax.scipy.signal.stft, {func}jax.scipy.signal.welch.

jax 0.3.0 (Feb 10, 2022)

jaxlib 0.3.0 (Feb 10, 2022)

  • Changes
    • Bazel 5.0.0 is now required to build jaxlib.
    • jaxlib version has been bumped to 0.3.0. Please see the design doc for the explanation.

jax 0.2.28 (Feb 1, 2022)

  • GitHub commits.
    • jax.jit(f).lower(...).compiler_ir() now defaults to the MHLO dialect if no dialect= is passed.
    • The jax.jit(f).lower(...).compiler_ir(dialect='mhlo') now returns an MLIR ir.Module object instead of its string representation.

jaxlib 0.1.76 (Jan 27, 2022)

... (truncated)

Commits
  • d1b6f5d Revert back to adding aval on Device buffers inside local_shards and convert ...
  • 1486be7 * Make _old_env thread local so that it can be used in multiple threads.
  • 8cb1692 Allow unevenly partitioned sharding_constraints.
  • 1baa59c Merge pull request #9594 from oliverdutton:boolean_jax2tf
  • 607e703 Turn execute_replicated into a class so we can access its fields.
  • d123a10 Merge pull request #9631 from gnecula:tf_doc
  • e6d94e1 docs: fix explanation
  • f926d08 docs: spelling mistake fixed
  • 3576388 [jax2tf] Added more links to the documentation
  • 20635e4 fix: simplify reduce_min with reduce_all
  • 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 2 years ago

Superseded by #274.