accelerate

baafaf4a - Fix the rng states of sampler's generator to be synchronized for correct sharding of dataset across GPUs (#2694)

Fix the rng states of sampler's generator to be synchronized for correct sharding of dataset across GPUs (#2694) * Fix the rng states of sampler's generator to be synchronized for correct sharding of dataset across GPUs * add tests