Egor Konovalov

Outdated and Broken Documentation: Module & Pytree Guide

3

## Description The [Module & Pytree guide](https://flax.readthedocs.io/en/stable/guides/pytree.html) in the Flax documentation is severely outdated and non-functional. Approximately half of the code cells in this guide fail to execute, making it...