flax

228c50e5 - Merge pull request #5379 from mohsinm-dev:fix-rngs-dunder-getattr