@Frameworks Engineer
I can confirm that switching to from tensorflow.keras.optimizers.legacy import Adam fixes the XLA problem, and TF 2.11 works fine.
So no need to downgrade the tensorflow version.
Thank you!
Topic:
Machine Learning & AI
SubTopic:
General
Tags: