Reply to Keras with tensorflow-metal freezes during training with image augmentation I got the same issue when use tfp.mcmc to sample from some distribution. Machine Learning & AI General Jan ’24