Let Flax-Orbax to not port the shape of target
arrays when they port the target
shardings.
#9843
The logs for this run have expired and are no longer available.
Loading