Method shuffle

mlx/data/Stream.cpp:131–133

}

Stream Stream::shuffle(int64_t buffer_size) const {
  return Stream(std::make_shared<stream::Shuffle>(self_, buffer_size));
}
Stream Stream::shuffle_if(bool cond, int64_t buffer_size) const {
  if (cond) {
    return shuffle(buffer_size);