search

ucl-bug / jaxdf

A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations
GNU Lesser General Public License v3.0
118 stars 7 forks source link

Bump jax from 0.3.13 to 0.3.14 in /.requirements #70

Closed dependabot[bot] closed 2 years ago

dependabot[bot] commented 2 years ago

Bumps jax from 0.3.13 to 0.3.14.

Changelog

Sourced from jax's changelog.

jax 0.3.14 (June 27, 2022)

  • GitHub commits.
  • Breaking changes
    • {func}jax.experimental.compilation_cache.initialize_cache does not support max_cache_size_ bytes anymore and will not get that as an input.
    • JAX_PLATFORMS now raises an exception when platform initialization fails.
  • Changes
    • Fixed compatibility problems with NumPy 1.23.
    • {func}jax.numpy.linalg.slogdet now accepts an optional method argument that allows selection between an LU-decomposition based implementation and an implementation based on QR decomposition.
    • {func}jax.numpy.linalg.qr now supports mode="raw".
    • pickle, copy.copy, and copy.deepcopy now have more complete support when used on jax arrays ({jax-issue}[#10659](https://github.com/google/jax/issues/10659)). In particular:
      • pickle and deepcopy previously returned np.ndarray objects when used on a DeviceArray; now DeviceArray objects are returned. For deepcopy, the copied array is on the same device as the original. For pickle the deserialized array will be on the default device.
      • Within function transformations (i.e. traced code), deepcopy and copy previously were no-ops. Now they use the same mechanism as DeviceArray.copy().
      • Calling pickle on a traced array now results in an explicit ConcretizationTypeError.
    • The implementation of singular value decomposition (SVD) and symmetric/Hermitian eigendecomposition should be significantly faster on TPU, especially for matrices above 1000x1000 or so. Both now use a spectral divide-and-conquer algorithm for eigendecomposition (QDWH-eig).
    • {func}jax.numpy.ldexp no longer silently promotes all inputs to float64, instead it promotes to float32 for integer inputs of size int32 or smaller ({jax-issue}[#10921](https://github.com/google/jax/issues/10921)).
    • Add a create_perfetto_link option to {func}jax.profiler.start_trace and {func}jax.profiler.start_trace. When used, the profiler will generate a link to the Perfetto UI to view the trace.
    • Changed the semantics of {func}jax.profiler.start_server(...) to store the keepalive globally, rather than requiring the user to keep a reference to it.
    • Added {func}jax.random.generalized_normal.
    • Added {func}jax.random.ball.
    • Added {func}jax.default_device.
    • Added a python -m jax.collect_profile script to manually capture program traces as an alternative to the Tensorboard UI.
    • Added a jax.named_scope context manager that adds profiler metadata to Python programs (similar to jax.named_call).
    • In scatter-update operations (i.e. :attr:jax.numpy.ndarray.at), unsafe implicit dtype casts are deprecated, and now result in a FutureWarning. In a future release, this will become an error. An example of an unsafe implicit cast is jnp.zeros(4, dtype=int).at[0].set(1.5), in which 1.5 previously was silently truncated to 1.
    • {func}jax.experimental.compilation_cache.initialize_cache now supports gcs bucket path as input.
    • Added {func}jax.scipy.stats.gennorm.

... (truncated)

Commits
  • 5b576cb Revert: Drop flatbuffers as a Python dependency of JAX.
  • 5b865ed Add the assume_metadata option to avoid waiting on ts.open which is very ...
  • a2f1aee Merge pull request #11248 from jakevdp:remove-jaxtestcase
  • 887abbc jax.test_util: remove deprecated test classes.
  • 997beb3 Merge pull request #11273 from hawkinsp:release
  • 1e29b7b Update CHANGELOG.md and setup.py for 0.3.14 release.
  • 0260360 Merge pull request #11244 from hawkinsp:xla
  • f4ddd3e Update XLA.
  • efefeac Drop flatbuffers as a Python dependency of JAX.
  • 93f5113 Merge pull request #11250 from LenaMartens:changelist/456788427
  • 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)
codecov[bot] commented 2 years ago

Codecov Report

Merging #70 (fb91e68) into main (28e0016) will increase coverage by 62.20%. The diff coverage is n/a.

@@            Coverage Diff             @@
##            main      #70       +/-   ##
==========================================
+ Coverage   0.53%   62.74%   +62.20%     
==========================================
  Files         13       13               
  Lines        934      934               
==========================================
+ Hits           5      586      +581     
+ Misses       929      348      -581
Impacted Files Coverage Δ
jaxdf/util.py 20.00% <0.00%> (+20.00%) :arrow_up:
jaxdf/ode.py 26.19% <0.00%> (+26.19%) :arrow_up:
jaxdf/operators/functions.py 37.14% <0.00%> (+37.14%) :arrow_up:
jaxdf/operators/differential.py 47.59% <0.00%> (+47.59%) :arrow_up:
jaxdf/operators/magic.py 67.07% <0.00%> (+67.07%) :arrow_up:
jaxdf/operators/linear_algebra.py 75.00% <0.00%> (+75.00%) :arrow_up:
jaxdf/core.py 78.62% <0.00%> (+75.57%) :arrow_up:
jaxdf/__init__.py 100.00% <0.00%> (+80.00%) :arrow_up:
jaxdf/geometry.py 81.96% <0.00%> (+81.96%) :arrow_up:
jaxdf/discretization.py 82.03% <0.00%> (+82.03%) :arrow_up:
... and 3 more

Continue to review full report at Codecov.

Legend - Click here to learn more Δ = absolute <relative> (impact), ø = not affected, ? = missing data Powered by Codecov. Last update 28e0016...fb91e68. Read the comment docs.