Version 0.6.3

[IvyZX]

What's Changed

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

New Contributors

• @yhtang made their first contribution in example/imagenet: use absolute path to locate the Flax root dir #2630

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

---

This discussion was created from the release Version 0.6.3.