Skip to content

Version 0.3.6

Choose a tag to compare

View all tags
@marcvanzee marcvanzee released this 27 Oct 21:00
· 3633 commits to main since this release
v0.3.6
136f41a
This commit was created on GitHub.com and signed with GitHub’s verified signature. The key has expired.
GPG key ID: 4AEE18F83AFDEB23
Expired
Verified
Learn about vigilant mode.

Breaking changes:

  • Move flax.nn to flax.deprecated.nn.

New features:

  • Add experimental checkpoint policy argument. See flax.linen.checkpoint
  • Add lifted versions of jvp and vjp.
  • Add lifted transformation for mapping variables. See flax.linen.map_variables.
Assets 2
Loading