flax

e8e0b2df - Merge pull request #3742 from chiamp:nnx_dataclass