jax

1829a667 - Merge pull request #21268 from jakevdp:register-dataclass