flax

4dd61948 - Fix PyTreeNode + Generic losing __parameters__ when Generic is last in bases (#5233)