Fix data distribution for model training in Jax Training (#23003)
* fix * Addressed automatic comments * Added tst * Fixed test
V
Vadym Doroshenko committed
b963d40136775917c358614dc3228db4b99400a5
Parent: 341ab04
Committed by GitHub <noreply@github.com>
on 5/28/2026, 10:56:49 PM