looks like the same issue as this one: https://developer.apple.com/forums/thread/691917
Seems Adam (and some other optimizers) currently do not work with tf-macos and tf-metal specially on Intel/AMD Macs.
jzhangc
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
Toronto, ON
Post
Replies
Boosts
Views
Activity
Another workaround is to use RectifiedAdam optimizer from tensorflow-addon, if SGD does not work as well.