jax

18f4456f - #sdy Add dataclass that stores the data needed to build a Shardy MLIR sharding attribute.