jax

ed849ff9 - Make sure to call the superclass' __init__() on a newly created instance in PositionalSharding._remake().

Make sure to call the superclass' __init__() on a newly created instance in PositionalSharding._remake(). If we don't do this, the C++ base class is left in an uninitialized state, leading to failures elsewhere in the test suite. PiperOrigin-RevId: 673411282