flax

Fix PyTreeNode + Generic losing __parameters__ when Generic is last in bases

#5237

Merged