Version 0.6.3

What's Changed

• Add gfile api shim to remove tensorflow dependency for basic IO. by @chiamp in #2586
• Remove Mypy type errors by @zaxtax in #2594
• Attempt to fix pytype issue. by @levskaya in #2628
• example/imagenet: use absolute path to locate the Flax root dir by @yhtang in #2630
• Move Flax - The Sharp Bits ToC location by @8bitmp3 in #2633
• Speed up Github Actions CI by @levskaya in #2635
• Update requirements by @marcvanzee in #2652
• Switch to Orbax for Flax single-checkpoint support under the hood. by @copybara-service in #2637
• Added path discrepancy details in serialization errors by @chiamp in #2632
• Fix get_type_hints again by @cgarciae in #2654
• BatchNorm guide by @cgarciae in #2536
• Generalize pool to handle multiple batch dimensions by @cgarciae in #2591
• Improve transfer learning guide by @cgarciae in #2595
• update docstrings and error messages in traverse_util.py. by @copybara-service in #2666
• Use a different rng key for each batch element in DenseGeneral init by @j-towns in #2665
• Added check for Mac M1 chip, when conducting serialization test. by @chiamp in #2657
• Added explicit warning if flax.io is using default Python I/O operations by @chiamp in #2625
• Check file or directory before removing checkpoints. by @IvyZX in #2676
• updated extracting_intermediates in flax docs by @chiamp in #2685
• updated model_surgery in flax_docs by @chiamp in #2687
• updated getting_started in flax docs by @chiamp in #2684
• Updated flax docs. by @chiamp in #2667
• updated flax_basics in flax docs by @chiamp in #2686
• Update python version support by @cgarciae in #2682
• add mypy.ini placeholder. by @copybara-service in #2693
• Release 0.6.3 by @IvyZX in #2705

New Contributors

• @yhtang made their first contribution in #2630

Full Changelog: v0.6.2...v0.6.3