flax

d23a5779 - Merge pull request #1630 from jheek:improve-dataclass-doc