Hi all,
Dropping the tensorflow dependency version is not a good idea I'd assume.
Consider using legacy optimisers, as suggested by Apple above, In my case it solved the issue.
Dropping the version will fix the problem but comes with other problems such as https://developer.apple.com/forums/thread/708154