Hi there, I am trying to augment some image data in TensorFlow. I am running TensorFlow-macos version 2.6.0 and Tensorflow-metal version 0.2.0. When I run the following lines of code: data_augmentation = keras.Sequential( [ layers.RandomFlip("horizontal"), layers.RandomRotation(0.1), layers.RandomZoom(0.2), ] ) for images, _ in train_dataset.take(1): for i in range(9): augmented_images = data_augmentation(images) I get the following error message: NotFoundError: No registered 'RngReadAndSkip' OpKernel for 'GPU' devices compatible with node {{node RngReadAndSkip}} . Registered: device='XLA_CPU_JIT' device='CPU' [Op:RngReadAndSkip] I have searched the internet extensively for a workaround, however have not found one. Any help would be greatly appreciated. Edie