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
wbartkowski
Users receive reputation points by contributing quality content to the forums.
Posts include post and replies published on the forums.
Post authors can mark replies as Accepted to indicate successful solutions.
Apple Developer Forums admins can mark replies as Apple Recommended to indicate an approved solution
From
Warsaw, Poland
Post
Replies
Boosts
Views
Activity
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
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