Version 0.4.1

andsteing released this 23 Mar 14:33

· 2306 commits to main since this release

ad8dfe0

What's Changed

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.

Assets 2