Demux

Any particular reason why you import shuffle instead of using np.random.shuffle?