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.2.12 #172

Closed dependabot[bot] closed 3 years ago

dependabot[bot] commented 3 years ago

Bumps jax from 0.2.8 to 0.2.12.

Changelog

Sourced from jax's changelog.

jax 0.2.12 (April 1 2021)

  • GitHub commits.
  • New features
    • New profiling APIs: {func}jax.profiler.start_trace, {func}jax.profiler.stop_trace, and {func}jax.profiler.trace
    • {func}jax.lax.reduce is now differentiable.
  • Breaking changes:
    • The minimum jaxlib version is now 0.1.64.
    • Some profiler APIs names have been changed. There are still aliases, so this should not break existing code, but the aliases will eventually be removed so please change your code.
      • TraceContext --> {func}~jax.profiler.TraceAnnotation
      • StepTraceContext --> {func}~jax.profiler.StepTraceAnnotation
      • trace_function --> {func}~jax.profiler.annotate_function
    • Omnistaging can no longer be disabled. See omnistaging for more information.
    • Python integers larger than the maximum int64 value will now lead to an overflow in all cases, rather than being silently converted to uint64 in some cases ({jax-issue}[#6047](https://github.com/google/jax/issues/6047)).
    • Outside X64 mode, Python integers outside the range representable by int32 will now lead to an OverflowError rather than having their value silently truncated.
  • Bug fixes:
    • host_callback now supports empty arrays in arguments and results ({jax-issue}[#6262](https://github.com/google/jax/issues/6262)).
    • {func}jax.random.randint clips rather than wraps of out-of-bounds limits, and can now generate integers in the full range of the specified dtype ({jax-issue}[#5868](https://github.com/google/jax/issues/5868))

jaxlib 0.1.65 (unreleased)

jax 0.2.11 (March 23 2021)

  • GitHub commits.

  • New features:

    • #6112 added context managers: jax.enable_checks, jax.check_tracer_leaks, jax.debug_nans, jax.debug_infs, jax.log_compiles.
    • #6085 added jnp.delete

  • Bug fixes:

    • #6136 generalized jax.flatten_util.ravel_pytree to handle integer dtypes.
    • #6129 fixed a bug with handling some constants like enum.IntEnums
    • #6145 fixed batching issues with incomplete beta functions
    • #6014 fixed H2D transfers during tracing
    • #6165 avoids OverflowErrors when converting some large Python integers to floats

  • Breaking changes:

    • The minimum jaxlib version is now 0.1.62.

... (truncated)

Commits
  • 8691244 Merge pull request #6319 from skye:version
  • 7b42011 Update jax version and changelog
  • 72a75ee Merge pull request #6315 from jakevdp:changelog
  • 7c4e26a Rename GitHub workflow "without omnistaging" to "with 3.7"
  • 2a091d2 Update changelog for #5868
  • 9ca11d4 Merge pull request #5868 from jakevdp:fix-randint
  • 7a5d1b6 Merge pull request #6311 from minoring:fix-minor-format-lax-numpy-test
  • 94e2c99 Add an api_util hook facility, and add a hook to jax2tf.convert.
  • d4739f2 Fix minor indentation format in lax_numpy_test
  • 2090431 random.randint: support generating the full range of dtype
  • 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

Superseded by #184.