Thank you very much for such a quick reply. That's very strange, especially as it works fine on macOS 12.0.1.

thank you a lot for workaround, but I hope next versions of Metal PluggableDevice will address this issue

Hi. I have reported this issue more that month ago. You can find in comments to my post that tf.random wrong behaviour makes this problem science MacOS 12.1. https://developer.apple.com/forums/thread/696835