Version 0.6.8

Changelog

• The automatic checkpoint migration was temporarily rolled back due to legacy compatibility issues.
  • We still recommend you to use the upgrade guide and migrate completely to the Orbax API to ensure stability.
  • Or alternatively, add flax.config.update('flax_use_orbax_checkpointing', True) to your project to avoid being impacted by the automatic migration process.
• Added utility functions to frozen_dict api.
• Migrated Flax away from register_keypaths.
• Fixes kwargs in convert_to_graphs_tuple_fn.
• Fixed examples in a few ways:
  • Bumped the TF version
  • Used latest checkpoint formats
  • Other misc fixes.

List of PRs

• Add Google Research examples with Flax by @8bitmp3 in #2934
• [Trivial] [Post-release] Bump up the HEAD version number by @IvyZX in #2946
• fix tabulate duplicating parameters by @cgarciae in #2931
• permit passing mesh or rules to spmd primitives by @cgarciae in #2939
• fixed ScopeParamShapeError message by @chiamp in #2947
• Fix checkpointing docs for Orbax 0.1.4 by @IvyZX in #2962
• Migrate Flax away from register_keypaths by @copybara-service in #2963
• set axis='batch' by @yechengxi in #2967
• Add Bidirectional by @cgarciae in #2770
• Bump tensorflow from 2.9.3 to 2.11.1 in /examples/ppo by @dependabot in #2984
• Bump tensorflow from 2.9.3 to 2.11.1 in /examples/imagenet by @dependabot in #2983
• Bump tensorflow from 2.9.3 to 2.11.1 in /examples/sst2 by @dependabot in #2980
• Bump tensorflow from 2.9.3 to 2.11.1 in /examples/nlp_seq by @dependabot in #2979
• Bump tensorflow from 2.9.3 to 2.11.1 in /examples/lm1b by @dependabot in #2977
• Adds missing api references by @cgarciae in #2926
• Added utility functions to frozen_dict api by @chiamp in #2975
• Fixed pop utility fn doc string by @chiamp in #2958
• release 0.6.8 by @IvyZX in #2987

Full Changelog: v0.6.7...v0.6.8