Good point tf.random.set_seed(), but this is not a solutions for calling tf.random. if you are using annotation @tf.function , at least for me in case and configuration as I described it here

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

Hi. I have reported this issue more that month ago. You can find in comments to my post that tf.random wrong behavior makes this problem. It was ok before upgrade to macOS 12.1. On 12.2 beta 2 also doesn't work. https://developer.apple.com/forums/thread/696835 And there's also a great analysis of this issue here https://developer.apple.com/forums/thread/697057