Hi, thanks for the suggestion - I was able to follow the same procedure in v2.12 and it is running, albeit with the known slowdown - thanks! I had some trouble in v2.13 as it looks like they have restructured the code a little. I also tried to implement a legacy-style version of AdamW based on the Adam class that already exists, but unfortunately this also ground to a halt because there is a point in which the python class calls an ApplyAdam function that is compiled in C++ (I can modify C++ but don't fancy trying to modify and recompile tensorflow!) I would also be happy to help contribute to a more durable solution - the tf forum directed me back here, let's see if your keras issue has any luck - I am not 100% sure who is in charge of tf/apple compatibility

Update: issue is resolved in the tf-macos nightly (see https://github.com/tensorflow/tensorflow/issues/60652)