At least for me it works if I do with tf.device('/cpu:0'): tf.random.uniform((10,)) But of course that's not a perfect solution. Also tf.sort and tf.argsort are broken. You can use the same workaround for them.

Persists in tensorflow-metal==0.3.0

Still persists in tensorflow 2.6.0.