flax

f265d914 - Merge pull request #4747 from vfdev-5:fix-docs-eval_shape