Skip to content

Let Flax-Orbax to not port the shape of target arrays when they port the target shardings. #9838

Let Flax-Orbax to not port the shape of target arrays when they port the target shardings.

Let Flax-Orbax to not port the shape of target arrays when they port the target shardings. #9838