Skip to content

Update flax requirement from <0.5,>=0.4 to >=0.4,<0.6

Vicentini Filippo requested to merge dependabot/pip/flax-gte-0.4-and-lt-0.6 into master

Created by: dependabot[bot]

Updates the requirements on flax to permit the latest version.

Release notes

Sourced from flax's releases.

Version 0.5.0

New features:

  • Added flax.jax_utils.ad_shard_unpad() by @​lucasb-eyer
  • Implemented default dtype FLIP. This means the default dtype is now inferred from inputs and params rather than being hard-coded to float32. This is especially useful for dealing with complex numbers because the standard Modules will no longer truncate complex numbers to their real component by default. Instead the complex dtype is preserved by default.

Bug fixes:

  • Fix support for JAX's experimental_name_stack.

Breaking changes:

  • In rare cases the dtype of a layer can change due to default dtype FLIP. See the "Backward compatibility" section of the proposal for more information.
Changelog

Sourced from flax's changelog.

0.5.0

  • Added flax.jax_utils.ad_shard_unpad() by @​lucasb-eyer
  • Implemented default dtype FLIP. This means the default dtype is now inferred from inputs and params rather than being hard-coded to float32. This is especially useful for dealing with complex numbers because the standard Modules will no longer truncate complex numbers to their real component by default. Instead the complex dtype is preserved by default.

Bug fixes:

  • Fix support for JAX's experimental_name_stack.

Breaking changes:

  • In rare cases the dtype of a layer can change due to default dtype FLIP. See the "Backward compatibility" section of the proposal for more information.

0.4.2

New features:

  • Add lifted conditional nn.cond.
  • Improved error messages: parameters not found, loading checkpoints.
  • Replace jax.tree_multimap (deprecated) with jax.tree_map.
  • Add the "Module Lifecycle" design note.
  • Add support for JAX dynamic stack-based named_call

Bug fixes:

  • Handle rate==1.0 edgecase in Dropout.
  • Fix bug where Linen Module state is reused.
  • Bug fixes and generalizations of nn.partitioning API.

0.4.1

New features:

  • Added locally-connected (unshared CNN) layer flax.linen.ConvLocal.
  • Improved seq2seq example: Factored our model and input pipeline code.
  • Added Optax update guide and deprecated flax.optim.
  • Added sep argument to flax.traverse_util.flatten_dict().
  • Implemented Sequential module, in flax.linen.combinators.

0.4.0

Breaking changes:

  • flax.deprecated.nn is removed. Please pin to flax==0.3.6 if you are still using it.
  • PixelCNN++ example is removed. It was not working well on TPU.
  • linen Normalization layers no longer downcast double and complex floats tofloat32 when computing the mean and variance.

New features:

  • Added flax.linen.custom_vjp for custom derivatives inside a Module.

... (truncated)

Commits

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)

Merge request reports